节点大小取决于NetworkX上的节点度
问题描述
我以.json文件的形式将我的Facebook数据导入我的计算机。数据的格式为:
{"nodes":[{"name":"Alan"},{"name":"Bob"}],"links":[{"source":0,"target:1"}]}
然后,我使用以下函数:
def parse_graph(filename):
"""
Returns networkx graph object of facebook
social network in json format
"""
G = nx.Graph()
json_data=open(filename)
data = json.load(json_data)
# The nodes represent the names of the respective people
# See networkx documentation for information on add_* functions
G.add_nodes_from([n['name'] for n in data['nodes']])
G.add_edges_from([(data['nodes'][e['source']]['name'],data['nodes'][e['target']]['name']) for e in data['links']])
json_data.close()
return G
要使此.json文件能够在NetworkX上使用图形。如果我找到节点的度数,我知道如何使用的唯一方法是:
degree = nx.degree(p)
其中p是我所有朋友的图表。现在,我想绘制该图,使节点的大小与该节点的度数相同。我该怎么做?
使用:
nx.draw(G,node_size=degree)
不起作用,我想不出其他方法。
解决方案
针对使用Networkx 2.x的用户的更新
接口已从v1.x改为v2.x。networkx.degree
不再返回dict
,而是根据documentation返回DegreeView
对象。
有从1.x迁移到2.x的指南here。
在这种情况下,基本上归结为使用dict(g.degree)
而不是d = nx.degree(g)
。
更新后的代码如下所示:
import networkx as nx
import matplotlib.pyplot as plt
g = nx.Graph()
g.add_edges_from([(1,2), (2,3), (2,4), (3,4)])
d = dict(g.degree)
nx.draw(g, nodelist=d.keys(), node_size=[v * 100 for v in d.values()])
plt.show()
nx.der(P)返回一个dict,而node_size keywod argument需要一个标量或一个大小数组。您可以按如下方式使用dict nx.der返回:
import networkx as nx
import matplotlib.pyplot as plt
g = nx.Graph()
g.add_edges_from([(1,2), (2,3), (2,4), (3,4)])
d = nx.degree(g)
nx.draw(g, nodelist=d.keys(), node_size=[v * 100 for v in d.values()])
plt.show()
相关文章