小编Viv*_*P20的帖子

Networkx:如何从csv文件创建图形边缘?

我正在尝试使用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)

python csv graph networkx python-2.7

5
推荐指数
1
解决办法
7515
查看次数

标签 统计

csv ×1

graph ×1

networkx ×1

python ×1

python-2.7 ×1