Networkx边缘颜色不一致

2024-06-28 19:38:38 发布

您现在位置:Python中文网/ 问答频道 /正文

嗨,我正在尝试使用networkx来绘制神经网络的数据。 神经元连接可以是兴奋性的,也可以是抑制性的。我用from/to/value列创建了一个数据帧。值是兴奋性或抑制性的分类变量编码。 然而,我有一些不一致之处。这里我展示了神经元n°245,如果我绘制整个数据,与如果我只绘制

df = df[(df['from'] == 245) | (df['to'] == 245)]

使用完整数据集

仅编号245

正如你所看到的,有些边缘的颜色不同。。。比如N245-N234

以下是我正在使用的代码:

df = pd.DataFrame({'from':pre_syn,'to':post_syn,'value':type_syn})
df['value']=pd.Categorical(df['value'])
df = df[(df['from'] == 245) | (df['to'] == 245)]

G=nx.from_pandas_edgelist(df, 'from', 'to', create_using=nx.DiGraph() )
pos=nx.spring_layout(G)
nx.draw(G,pos,connectionstyle='arc3, rad = 0.3', with_labels=True, node_size=500, edge_color=df['value'].cat.codes,
        width=1.0, edge_cmap=plt.cm.Set1)

非常感谢你的帮助


Tags: to数据fromposnetworkxdfvalue绘制