python生成环形树状图-networkx
一，项目简介：
利用networkx官方中的CircularTree案例，修改节点的名称，利用唐代诗人姓名与其诗作中的关键字生成的边，形成以诗人关键字为代表的三层环形树状图。
附上原官方的CircularTree图，以及官方代码的链接。
https://networkx.github.io/documentation/stable/auto_examples/drawing/plot_circular_tree.html#sphx-glr-auto-examples-drawing-plot-circular-tree-py
二，实现方法：
1，从全唐诗文本中提取指定诗人的作品，我找到的版本标题含有中文【】,对于处理标题相当有利。
2，逐字解析指定诗人作品中的字，并按出现频率的高低排序，提取出现频率最高的m个字作为关键字（关键字不宜太多）；
3，使用关键字搜索指定诗人的作品，将关键字前后的字+关键字组成的词语提取出来，形成列表，并按出现的频率排序，提取m个；
4，将m个关键字与m*m个词语组合成元组，形成networkx中的边；
5，引入networkx画关系图，使用balanced_tree和graphviz_layout组合形成节点名称+位置的字典，替换节点名称为诗人名称+关键字+词语的节点列表；
6，生成圆形树状图。
三，使用到的内置库和第三方库
1，collections中的Counter，用来统计列表中字和词语的个数，生成一个由字词语为键，出现的频率为值的字典；
2，re库，统计关键字在文本中出现的位置k，为提取(k-1)+k和k+(k+1)做必要的准备；
3，networkx中的balanced_tree生成指定的节点，graphviz_layout生成节点信息；Graph和draw生成图形；
4，matplot.pyplot输出图形。
四，附上几张生成的效果，就放李杜白元的作品吧，毕竟他们的作品在全唐诗中出现的比例也是最高的。
五，附上源代码。
from collections import Counter
import re
import networkx as nx
import matplotlib.pyplot as plt
def getTxtByWriter(fn,writer,stopword,n):#获取全唐诗中指定作者的诗
fo=open(fn,'r',encoding='utf-8-sig').readlines()
peomWithWriter=''#诗的内容+作者标题
for i in fo:
if '【' in i:
i=i.replace(i,'!'+i[i.find('【'):])#如果行有中文括号，表示此行为标题
peomWithWriter+=(i.strip('\n'))
else:
peomWithWriter+=(i.strip('\n'))
writerPoemWithIcon=''#诗的内容+标点符号
for i in peomWithWriter.split('!'):#使用split分段开成列表
if '】'+writer in i:#遍历列表是否有作者名称，有的话加入作者诗的字符串集
i=i.split('】')[1]
writerPoemWithIcon+=i.replace(writer,'')#replace('，','').replace('。','').replace(' ','')
return writerPoemWithIcon
def getKeyWord(fn,writer,stopword,n):#获取出现频率最高的字
writerPoemWithIcon=getTxtByWriter(fn,writer,stopword,n)
WordDict=Counter(list(writerPoemWithIcon))#将生成的字符串List成列表，并统计每个字出现的次数
sw=open(stopword,'r',encoding='utf-8').read().split('.')
keyWordList=[]#关键字列表
for k,v in WordDict.items():
if k not in sw and v>n:#如果元素不在stopword中且出现的次数超过n次，就可以加入列表中。
keyWordList.append(k)
return keyWordList
# print(getKeyWord('全唐诗.txt','杜甫','ChineseStopWord.txt',400))
def getConnectWord(fn,writer,stopword,n,m):#文本，作者，停用词，关键字出现的频率，前m个关键字
keyWordList=getKeyWord(fn,writer,stopword,n)[:m]#获取前m个关键字
writerPoemWithIcon=getTxtByWriter(fn,writer,stopword,n)
connectWordList=[]
kwl=[]
a=[writer]+keyWordList
wk=[]
for i in keyWordList:
wk.append((writer,i))
singleWord=[]
list=[j.start() for j in re.finditer(i,writerPoemWithIcon)]#list是关键字i出现位置的列表
for k in list:
if writerPoemWithIcon[k+1] not in '，。 ' and writerPoemWithIcon[k-1] not in '，。 ':
singleWord.append(writerPoemWithIcon[k:k+2])
singleWord.append(writerPoemWithIcon[k-1:k+1])
connectWord=[]
kw=[]
for k,v in sorted(Counter(singleWord).items(),key= lambda x:x[1],reverse=True):#关键字关联词的前m个
connectWord.append((i,k))
kw.append(k)
connectWordList+=connectWord[:m]
kwl=kwl+kw[:m]
return wk+connectWordList,a+kwl
# print(getConnectWord('全唐诗.txt','杜甫','ChineseStopWord.txt',400,6))
def createCircularTree(fn,writer,stopword,n,m):#生成树形分支图
connectWordList, nodeList = getConnectWord(fn, writer, stopword, n,m)
try:
import pygraphviz
from networkx.drawing.nx_agraph import graphviz_layout
except ImportError:
try:
import pydot
from networkx.drawing.nx_pydot import graphviz_layout
except ImportError:
raise ImportError("This example needs Graphviz and either "
"PyGraphviz or pydot")
plt.figure(figsize=(10, 10),facecolor='blue')
g=nx.Graph()
pos=graphviz_layout(nx.balanced_tree(m,3),prog='twopi')
g.add_edges_from(connectWordList)
p=dict(zip(nodeList,pos.values()))
d=dict(g.degree)
nx.draw(g,p,with_labels=True,nodelist=d.keys(),node_size=[v*400 for v in d.values()],node_color=range(len(nodeList)),
cmap=plt.cm.Blues,alpha=0.8)
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.savefig('%s诗的主要意象.png'% writer,dpi=100)