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