节点大小取决于NetworkX上的节点级别

Question

我以.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)

没有工作，我想不出另一种方法。

Answer 1

Update for those using networkx 2.x

API已从v1.x更改为v2.x. networkx.degree不再返回dict，而是根据DegreeView返回documentation对象。

有一个从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.degree（p）返回一个字典，而node_size keywod argument需要一个标量或一系列大小。您可以使用dict nx.degree返回如下：

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()

Answer 2

@ miles82提供了一个很好的答案。但是，如果您已经使用G.add_nodes_from(nodes)之类的东西将节点添加到图表中，那么我发现d = nx.degree(G)可能不会以与节点相同的顺序返回度数。

基于上一个答案，您可以稍微修改解决方案以确保度数的顺序正确：

d = nx.degree(G)
d = [(d[node]+1) * 20 for node in G.nodes()]

请注意d[node]+1，它将确保将零度节点添加到图表中。