networkx边缘颜色始终为黑色,忽略颜色属性

时间:2020-05-16 21:02:09

标签: python graph networkx edges

我要代表一些分析规则,并且我希望某些节点出入的边缘具有相同的颜色。我以不同的方式来处理它,但是没有一个起作用。例如:

G = nx.DiGraph()
color_map = []
color_iter = 0
edge_colors_iter = ["#4e79a9", "#59a14f", "#9c755f", "#f28e2b", "#edc948", "#bab0ac", "#e15759", "#b07aa1",
                    "#76b7b2", "#ff9da7"]

for index, row in rules.iterrows():

    edges_to_add = []
    color_of_rule = edge_colors_iter[color_iter]


    #left_side_items and right_side_items are essentially lists for in and out edges 
    #of the certain node i need with the same color
    # ...

    for item in left_side_items:
        edges_to_add.append((str(item), "R"+str(rule_id)))

    for item in right_side_items:
        edges_to_add.append(("R"+str(rule_id), str(item)))

    G.add_edges_from(edges_to_add, color=color_of_rule)

    color_iter += 1

for node in G:

    if str(node).startswith("R"):
        color_map.append('#f9dc4c')

    else:
        color_map.append('#339e34')

nx.draw_circular(G, node_color=color_map, with_labels=True)
plt.show()

我得到正确的图表,但是颜色总是黑色,例如 enter image description here

2 个答案:

答案 0 :(得分:1)

您可以在指定所需颜色的地方传递关键字参数edge_color
示例:

import networkx as nx
import matplotlib.pyplot as plt

G = nx.complete_graph(n=6)

color_map = []
for u, v in G.edges():
    if u % 2 == 0 and v % 2 == 0:
        color_map.append('#f9dc4c')
    else:
        color_map.append('#339e34')

nx.draw_circular(G, edge_color=color_map, with_labels=True)
plt.show()

答案 1 :(得分:0)

显然nx.draw_circular不考虑colors属性,必须明确给出。

for item in left_side_items:
    G.add_edge(str(item), "R"+str(rule_id), color=color_of_rule)

for item in right_side_items:
    G.add_edge("R"+str(rule_id), str(item), color=color_of_rule)

color_iter += 1

将颜色添加到每个边缘后,您可以将它们添加到将传递到draw_circular的列表中。

edges = G.edges()
colors = [G[u][v]['color'] for u, v in edges]

nx.draw_circular(G, edge_color=colors, node_size=final_node_sizes, node_color=color_map, with_labels=True)
plt.show()