我正在尝试使用networkx创建图,到目前为止,我已经从以下文本文件创建了节点:文件1(user_id.txt)样本数据:
user_000001
user_000002
user_000003
user_000004
user_000005
user_000006
user_000007
Run Code Online (Sandbox Code Playgroud)
文件2(user_country.txt)样本数据:如果用户未输入其国家/地区详细信息,也将包含少量空白行
Japan
Peru
United States
Bulgaria
Russian Federation
United States
Run Code Online (Sandbox Code Playgroud)
文件3(user_agegroup.txt)数据:包含四个年龄段
[12-18],[19-25],[26-32],[33-39]
Run Code Online (Sandbox Code Playgroud)
我还有另外两个文件,其中包含以下示例数据,用于在图形中添加边
文件4(id,agegroup.txt)
user_000001,[19-25]
user_000002,[19-25]
user_000003,[33-39]
user_000004,[19-25]
user_000005,[19-25]
user_000006,[19-25]
user_000007,[26-32]
Run Code Online (Sandbox Code Playgroud)
文件5(id,country.txt)
(user_000001,Japan)
(user_000002,Peru)
(user_000003,United States)
(user_000004,)
(user_000005,Bulgaria)
(user_000006,Russian Federation)
(user_000007,United States)
Run Code Online (Sandbox Code Playgroud)
到目前为止,我已经编写了以下代码来绘制仅包含节点的图形:(请检查代码,因为print g.number_of_nodes()
尽管print g.nodes()显示了正确的节点数,但从不打印正确的节点数。)
import csv
import networkx as nx
import matplotlib.pyplot as plt
g=nx.Graph()
#extract and add AGE_GROUP nodes in graph
f1 = csv.reader(open("user_agegroup.txt","rb"))
for row in f1:
g.add_nodes_from(row)
nx.draw_circular(g,node_color='blue')
#extract and add COUNTRY nodes in …Run Code Online (Sandbox Code Playgroud)