使用NetworkX在Python中可视化图形

时间:2016-02-15 19:43:26

标签: python numpy networkx

我试图在NetworkX中可视化图形。我需要像这样着色图形:中心节点需要着色为黑色。然后,更远的所有节点将需要变浅,但是当我运行代码时,我得到这个错误:

错误:无法转换参数类型< class' numpy.ndarray' >到行上的rgba数组:

$(document).ready(function(){

我认为问题在于:

  nx.draw_networkx_nodes(G,pos,nodelist=p.keys(),node_size=90,
  node_color=p.values(),cmap=plt.cm.Reds_r)

代码是:

node_color=p.values()

完整追溯

import numpy
import pandas
import networkx as nx
import unicodecsv as csv
import community
import matplotlib.pyplot as plt

# Generate the Graph
G=nx.davis_southern_women_graph()

# Create a Spring Layout
pos=nx.spring_layout(G)

# Find the center Node
dmin=1
ncenter=0
for n in pos:
   x,y=pos[n]
   d=(x-0.5)**2+(y-0.5)**2
   if d<dmin:
      ncenter=n
      dmin=d

""" returns a dictionary of nodes and their distance to the node 
supplied as an argument. We will then use these distances 
to determine colors"""
p=nx.single_source_shortest_path_length(G,ncenter)

plt.figure(figsize=(8,8))
nx.draw_networkx_edges(G,pos,nodelist=[ncenter],alpha=0.4)
nx.draw_networkx_nodes(G,pos,nodelist=p.keys(),node_size=90,
   node_color=p.values(),cmap=plt.cm.Reds_r)

plt.show()

1 个答案:

答案 0 :(得分:1)

错误在绘图节点的功能中。 必须将Set<Companies>值放入p.keys()nodelist的列表中,否则它将无效。

所以正确的行是:

node_color