4

我使用以下代码构建了一个图表：

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]]

然后我使用以下方法显示：

networkx.draw(G)
matplotlib.pyplot.show()

它显示得很好，除了所有边缘都是黑色的，而不是我在上面的代码片段中尝试分配的颜色。有任何想法吗？

4

3 回答 3

6

您可以在不循环绘制边缘的情况下做到这一点。对于较大的图形，它会更快。 

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()
于 2013-05-14T13:12:01.917 回答
1

嗯，终于找到我需要做的了。颜色显然需要分开，网络需要用draw_networkx. 所以上面的三个for循环要替换为：

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")
于 2010-10-16T22:07:54.827 回答
0

文档字符串显示：

edge_color ：颜色字符串，或浮点数组边缘颜色。可以是单个颜色格式字符串（默认='r'），也可以是长度与 edgelist 相同的颜色序列。如果指定了数值，它们将使用 edge_cmap 和 edge_vmin,edge_vmax 参数映射到颜色。

于 2013-05-13T14:45:27.733 回答