Размер узла зависит от степени узла в NetworkX

Question

Я импортировал свои данные в 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) 

Где р является графиком всех моих друзей. Теперь я хочу построить график таким образом, чтобы размер узла был таким же, как и степень этого узла. Как мне это сделать?

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

nx.draw(G,node_size=degree) 

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

Answer 1

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