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