Node размер зависит от степени Node от NetworkX

Я импортировал свои данные в Facebook на свой компьютер в виде файла .json. Данные находятся в формате:

{"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 - график всех моих друзей. Теперь я хочу построить график таким образом, чтобы размер node был таким же, как и степень этого node. Как это сделать?

Использование:

nx.draw(G,node_size=degree)

не работал, и я не могу придумать другого метода.

Ответ 1

nx.degree(p) возвращает dict, в то время как node_size аргумент keywod требует скаляр или массив размеров. Вы можете использовать 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()

Ответ 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, который будет уверен, что узлы нулевой степени добавляются в диаграмму.