我是Networkx(2.4版)的新手,对于尝试添加节点列表时遇到的stochastic_block_model错误,我有些困惑。我正在尝试使用此代码为网络中的每个块设置不同的颜色属性:
import networkx as nx
N_p = 10
N_n = 10
N_0 = 30
sizes = [N_p,N_n,N_0]
probs = [[0.25,0.05,0.02],[0.05,0.35,0.07],[0.02,0.07,0.40]]
nodelist = ['blue' for i in range(N_p)]
nodelist.extend(['red' for i in range(N_n)])
nodelist.extend(['green' for i in range(N_0)])
G = nx.stochastic_block_model(sizes,probs,nodelist=nodelist,seed=0,directed=1)
但是我收到以下错误消息:
...
/opt/anaconda3/lib/python3.7/site-packages/networkx/generators/community.py in stochastic_block_model(sizes,p,nodelist,seed,directed,selfloops,sparse)
576 raise nx.NetworkXException("'nodelist' and 'sizes' do not match.")
577 if len(nodelist) != len(set(nodelist)):
--> 578 raise nx.NetworkXException("nodelist contains duplicate.")
579 else:
580 nodelist = range(0,sum(sizes))
NetworkXException: nodelist contains duplicate.
我在做什么错了?