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