为networkx中的节点分配了错误的颜色

Question

我很难在网络中绘制正确的颜色。 这是我的数据集，其中用户是节点，值是链接到的用户。

    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等）（绿色）。

Answer 1

颜色本身没有错。您只是想为每个节点分配一种颜色，这很可能是边缘的着色。在这种情况下，您需要将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+')

输出