如何解决为networkx中的节点分配了错误的颜色 输出
我很难在网络中绘制正确的颜色。 这是我的数据集,其中用户是节点,值是链接到的用户。
User Val Color
92 Laura NaN red
100 Laura John red
148 Laura Mike red
168 Laura Mirk red
293 Laura Sara red
313 Laura Sim red
440 Martyn Pierre orange
440 Martyn Hugh orange
440 Martyn Lauren orange
440 Martyn Sim orange
我正在使用以下代码:
G = nx.from_pandas_edgelist(df,'User','Val')
labels = [i for i in dict(G.nodes).keys()]
labels = {i:i for i in dict(G.nodes).keys()}
colors = df[["User","Color"]].drop_duplicates()["Color"]
plt.figure(3,figsize=(30,50))
pos = nx.spring_layout(G)
nx.draw(G,node_color = colors,pos = pos)
net = nx.draw_networkx_labels(G,pos = pos)
我尝试将“颜色”列中的颜色分配给用户。我的预期输出将是:
Laura(节点)为红色,Martyn(节点)为橙色,其他(John,Mike,Mirk等)(绿色)。
解决方法
颜色本身没有错。您只是想为每个节点分配一种颜色,这很可能是边缘的着色。在这种情况下,您需要将node_color
关键字替换为edge_color
。
G = nx.from_pandas_edgelist(df,'User','Val')
colors = df[["User","Color"]].drop_duplicates()["Color"]
plt.figure(figsize=(3,3))
pos = nx.spring_layout(G)
nx.draw(G,edge_color = colors,pos = pos)
net = nx.draw_networkx_labels(G,pos = pos)
plt.show()
我正在与df
合作:
t = '''
User Val Color
92 Laura NaN red
100 Laura John red
148 Laura Mike red
168 Laura Mirk red
293 Laura Sara red
313 Laura Sim red
440 Martyn Pierre orange
440 Martyn Hugh orange
440 Martyn Lauren orange
440 Martyn Sim orange'''
from io import StringIO
df = pd.read_csv(StringIO(t),sep='\s+')
输出
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。