突出显示 NetworkX 中的某些节点/边缘 - 使用 zip() 的问题

Question

我能够使用 networkx 填充网络图。我的问题是当我想突出显示图形无法生成的路径（例如最短路径）时，它将在下面返回错误。nx.draw_networkx_edges(Gr,pos,edgelist=path_edges,edge_color='r',width=10)  File "/usr/local/lib/python3.6/site-packages/networkx/drawing/nx_pylab.py", line 578, in draw_networkx_edges    if not edgelist or len(edgelist) == 0:  # no edges!TypeError: object of type 'zip' has no len()我寻找解决方案，这是因为脚本在 python3 上运行，因此我收到此错误。解决方案之一是更改和添加如下列表。原来的：Gr = nx.DiGraph() edges = graphGr.add_edges_from(edges)pos = nx.spring_layout(Gr)path = nx.shortest_path(Gr,source=1,target=7)path_edges = zip(path,path[1:])nx.draw_networkx_nodes(Gr,pos,nodelist=path,node_color='r')nx.draw_networkx_edges(Gr,pos,edgelist=path_edges,edge_color='r',width=10)plt.axis('equal')plt.show()修改的：path = nx.shortest_path(Gr,source=1,target=7)path_edges = list(zip(path,path[1:]))nx.draw_networkx_nodes(Gr,pos,nodelist=path,node_color='r')nx.draw_networkx_edges(Gr,pos,edgelist=path_edges,edge_color='r',width=10)plt.axis('equal')plt.show()能够运行脚本没有错误并且能够生成图形，但突出显示的路径（红色）未与拓扑节点和链接对齐。红色路径应在顶部并与节点/路径 1-2-3-4-5-6-7 对齐。请参考下面的图片

Answer 1

我能够使用以下代码生成您似乎想要的图表 - 如果您遇到任何问题，请告诉我。您需要将zip对象转换为 a是正确的list，但我认为您的绘图代码中可能存在其他错误。nx.spring_layout如果您需要每次的输出都相同，则可以使用seed关键字参数，例如pos = nx.spring_layout(Gr, seed=123).

代码：

import networkx as nx

# Set up graph

Gr = nx.DiGraph() 

edges = [(i+1, i+2) for i in range(10)] + [(i+2, i+1) for i in range(10)]

Gr.add_edges_from(edges)

# Get position using spring layout

pos = nx.spring_layout(Gr)

# Get shortest path

path = nx.shortest_path(Gr,source=1,target=7)

path_edges = list(zip(path,path[1:]))

# Draw nodes and edges not included in path

nx.draw_networkx_nodes(Gr, pos, nodelist=set(Gr.nodes)-set(path))

nx.draw_networkx_edges(Gr, pos, edgelist=set(Gr.edges)-set(path_edges), connectionstyle='arc3, rad = 0.3')

# Draw nodes and edges included in path

nx.draw_networkx_nodes(Gr, pos, nodelist=path, node_color='r')

nx.draw_networkx_edges(Gr,pos,edgelist=path_edges,edge_color='r', connectionstyle='arc3, rad = 0.3')

# Draw labels

nx.draw_networkx_labels(Gr,pos)

输出：