使用draw_network_nodes()时，Python网络会更改节点的颜色 [英] Python networks change color of nodes when using draw_network_nodes()

问题描述

目标是获得类似的东西

要定义我使用的图形:

import matplotlib.pyplot as plt
import networkx as nx
graph = {
    '1': ['2', '3', '4'],
    '2': ['5','11','12','13','14','15'],
    '3' : ['6','7','66','77'],
    '5': ['6', '8','66','77'],
    '4': ['7','66','77'],
    '7': ['9', '10']
    } 

MG = nx.DiGraph()
MG.add_edges_from([(start, stop, {'weigth' :  len(graph[start]) }) 
     for start in graph for stop in graph[start]])

和绘制它的代码是:

plt.figure(figsize=(8,8))
pos=nx.graphviz_layout(MG,prog="twopi",root='1')
for n in MG.nodes_iter():
    nx.draw_networkx_nodes(MG, pos,
        nodelist = [n],
        node_size = 2000 / float(len(MG[n[0]])+1),
        node_color = (len(MG[n[0]])+1),
        alpha = 1/float(len(MG[n[0]])+1),
        with_labels=True
                       )
xmax=1.1*max(xx for xx,yy in pos.values())
ymax=1.1*max(yy for xx,yy in pos.values())
plt.xlim(0,xmax)
plt.ylim(0,ymax)
plt.show()

我想知道两件事:

• 如何根据每个节点具有的链接数(即len(MG[0]))来设置颜色节点?

• how can I have the color node depending on the number of links each node has (i.e. len(MG[0]))?

我的标签在哪里?

谢谢

推荐答案

这个怎么样?您可以使用matplotlib的颜色图将值映射到节点的颜色.

How about this? You can use matplotlib's colormaps to map values to colors for the nodes.

import matplotlib.pyplot as plt
import networkx as nx
graph = {
    '1': ['2', '3', '4'],
    '2': ['5','11','12','13','14','15'],
    '3' : ['6','7','66','77'],
    '5': ['6', '8','66','77'],
    '4': ['7','66','77'],
    '7': ['9', '10']
    }

MG = nx.DiGraph(graph)


plt.figure(figsize=(8,8))
pos=nx.graphviz_layout(MG,prog="twopi",root='1')

nodes = MG.nodes()
degree = MG.degree()
color = [degree[n] for n in nodes]
size = [2000 / (degree[n]+1.0) for n in nodes]

nx.draw(MG, pos, nodelist=nodes, node_color=color, node_size=size,
        with_labels=True, cmap=plt.cm.Blues, arrows=False)
plt.show()

这篇关于使用draw_network_nodes()时，Python网络会更改节点的颜色的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！