NetworkX: wie farbige Kanten zeichnen?
Frage
Ich habe ein Diagramm mit dem folgenden Code aufgebaut:
G = networkx.Graph()
G.add_edges_from([list(e) for e in P + Q + R])
colors = "bgrcmyk"
color_map = [colors[i] for i in range(n/2)]
# add colors
for i in range(len(P)):
edge = list(P[i])
G[edge[0]][edge[1]]['edge_color'] = color_map[i]
for i in range(len(P)):
edge = list(Q[perms[0][i]])
G[edge[0]][edge[1]]["color"] = color_map[perms[0][i]]
for i in range(len(P)):
edge = list(R[perms[1][i]])
G[edge[0]][edge[1]]["color"] = color_map[perms[1][i]]
, die ich mit dann angezeigt werden:
networkx.draw(G)
matplotlib.pyplot.show()
Es zeigt in Ordnung, außer dass alle Kanten in schwarz statt der Farben gefärbt sind, ich in diesem kleinen Skript zuweisen versucht. Irgendwelche Ideen?
Lösung 2
Hm, fand schließlich, was ich tun musste. Farben offenbar getrennt werden benötigt, und Netzwerk muss mit draw_networkx
gezogen werden. So sind die drei oben genannten for
Schleifen sind mit ersetzt werden:
pos=networkx.spring_layout(G)
for i in range(len(P)):
networkx.draw_networkx_edges(G,pos,
edgelist=[list(P[i]), list(Q[perms[0][i]]), list(R[perms[1][i]])],edge_color=color_map[i], width="8")
Andere Tipps
Sie können es tun, ohne Kanten in einer Schleife zu ziehen. Es wird schneller für größere Graphen.
import networkx as nx
import matplotlib.pyplot as plt
G = nx.Graph()
# example graph
for color in "bgrcmyk":
G.add_edge('s'+color,'t'+color, color=color)
# edge_color_attr = nx.get_edge_attributes(G,'color')
# edges = edge_color_attr.keys()
# colors = edge_color_attr.values()
edges,colors = zip(*nx.get_edge_attributes(G,'color').items())
nx.draw(G,edgelist=edges,edge_color=colors,width=10)
plt.show()
Die docstring zeigt:
edge_color: Farbe String oder Array von Schwimmern Kantenfarbe. Kann eine einzelne Farbe Format String (default = 'r'), oder eine Sequenz von Farben mit der gleichen Länge wie Kantenliste. Wenn numerische Werte angegeben werden, werden sie zu zugeordnet werden Farben mit dem edge_cmap und edge_vmin, edge_vmax Parameter.