[[189747]]

今天給大家?guī)砦乙粋€腳本，用來分析社會網(wǎng)絡(luò)關(guān)系。

這個圖我沒有用到gephi或者其他的工具，是我用python純腳本運(yùn)行出來的。簡單的實現(xiàn)了封裝，大家有興趣可以下載下腳本，運(yùn)行下。

原理知識

我就簡單說下原理吧，先刻畫一個簡單的圖A

import networkx as nx 
import matplotlib.pyplot as plt 
#有向圖 
DG = nx.DiGraph() 
#添加一個節(jié)點 
DG.add_node('A') 
#作圖，設(shè)置節(jié)點名顯示,節(jié)點大小，節(jié)點顏色 
nx.draw(DG,with_labels=True,node_size=900,node_color = 'green') 
plt.show() 

雙節(jié)點，有方向A–>B

#有向圖 
DG = nx.DiGraph() 
#添加一個節(jié)點 
DG.add_node('A') 
DG.add_node('B') 
#添加邊，有方向，A-->B 
DG.add_edge('A','B') 
#作圖，設(shè)置節(jié)點名顯示,節(jié)點大小，節(jié)點顏色 
nx.draw(DG,with_labels=True,node_size=900,node_color = 'green') 
plt.show() 

添加更多節(jié)點

import networkx as nx 
import matplotlib.pyplot as plt 
 
colors = ['red', 'green', 'blue', 'yellow'] 
#有向圖 
DG = nx.DiGraph() 
#一次性添加多節(jié)點，輸入的格式為列表 
DG.add_nodes_from(['A', 'B', 'C', 'D']) 
#添加邊，數(shù)據(jù)格式為列表 
DG.add_edges_from([('A', 'B'), ('A', 'C'), ('A', 'D'), ('D','A')]) 
#作圖，設(shè)置節(jié)點名顯示,節(jié)點大小，節(jié)點顏色 
nx.draw(DG,with_labels=True, node_size=900, node_color = colors) 
plt.show() 

好了，同樣的道理，咱們建立角色詞典，插入節(jié)點列表，然后遍歷插入有向邊關(guān)系，就能做出這圖：

《Python基于共現(xiàn)提取《釜山行》人物關(guān)系》

這篇文章寫的很好(希望不要黑我，我也是覺得真心好)。我基本上借鑒了這篇文章思路寫今天這個腳本，實現(xiàn)了自動生成關(guān)系網(wǎng)絡(luò)圖。

準(zhǔn)備工作

一、安裝matplotlib、networkx 
 
二、解決matplotlib無法寫中文問題 
 
1、找到pythonX\lib\site-packages\matplotlib\mpl-data\fonts\ttf文件夾 
 
2、matplotlib默認(rèn)調(diào)用的為DejaVuSans.ttf字體文件，網(wǎng)上下載個 
 
微軟雅黑.ttf 
 
3、將微軟雅黑文件名改為DejaVuSans.ttf粘貼到ttf文件夾下即可。 

腳本文件簡介

你下載后腳本文件夾名為：人民的名義

其中的relationship.py就是大鄧寫的庫(直接能調(diào)用的哦)

實現(xiàn)功能：

1、讀入小數(shù)數(shù)據(jù)和角色詞典后，對數(shù)據(jù)分詞后生成角色關(guān)系數(shù)據(jù)(有向關(guān)系數(shù)據(jù))

2、無需gephi繪圖即可制作絢麗角色關(guān)系網(wǎng)絡(luò)圖。

注意：

1、運(yùn)行腳本前，文件夾中只保留角色名單.txt 人民的名義.txt 和 relationship.py

2、要想圖片好看點，一定要聯(lián)網(wǎng)啊。里面我寫了個爬蟲，爬取一個配色網(wǎng)站的配色值。

如果你們想單機(jī)也可以畫出好看的圖，可以修改下代碼，將代碼爬蟲部分改成讀取本地顏色數(shù)據(jù)

使用示例

1、建好小說數(shù)據(jù)中的角色字典，格式如下

2、在文件夾中放入小說txt文件(人民的名義.txt)

3、在腳本文件夾中新建一個test.py文件

#導(dǎo)入relationship庫中的Relationship類 
from relationship import Relationship 
 
#自定義節(jié)點詞典（小說中人物角色） 
dictpath = r'/Users/suosuo/Desktop/人民的名義/角色名單.txt' 
#小說路徑，只能是編碼方式為utf-8的txt文件 
datapath = r'/Users/suosuo/Desktop/人民的名義/人民的名義.txt' 
#程序運(yùn)行生成的角色關(guān)系圖保存地址 
pic = r'/Users/suosuo/Desktop/人民的名義/人物關(guān)系圖.png' 
Re = Relationship(dictpath, datapath) 
relation = Re.relationship() 
graph = Re.network_digraph(relation, pic) 

4、運(yùn)行

文件夾中生成了 人物關(guān)系圖.png、node_edge.txt和node_freq.txt文件。

node_edge.txt 有向圖關(guān)系數(shù)據(jù)，可以后續(xù)導(dǎo)入gephi軟件自定義制圖

node_freq.txt 節(jié)點出現(xiàn)頻率

注意：每次運(yùn)行前請把人物關(guān)系圖.png、node_edge.txt和node_freq.txt文件刪除掉，再運(yùn)行。