NetworkX：从 shapefile 向图形添加边

Question

给定以下测试 shapefile，它仅由折线组成：

我能够重现 shapefile 中表示的空间网络的节点：

import networkx as nx
import matplotlib.pyplot as plt

G=nx.read_shp('C:\Users\MyName\MyFolder\TEST.shp') #Read shapefile as graph
pos = {k: v for k,v in enumerate(G.nodes())} #Get the node positions based on their real coordinates
X=nx.Graph() #Empty graph
X.add_nodes_from(pos.keys()) #Add nodes preserving real coordinates
nx.draw_networkx_nodes(X,pos,node_size=100,node_color='r')
plt.xlim(450000, 470000)
plt.ylim(430000, 450000)

基本上我使用了一个临时图 G 来提取最终作为图的一部分出现的节点的位置 X。这似乎工作得很好。

我的问题： 遵循使用 G 从 shapefile 中提取信息的相同想法，我如何绘制边缘？

如果我这样做

X.add_edges_from(pos.keys())

然后我得到这个错误，指向上面的行：

TypeError: object of type 'int' has no len()

Answer 1

补充一下我的评论： nx.read_shp() 也保存边缘信息。图 G 的节点看起来像 (x,y)。 draw_networkx_* 的 pos 参数需要是一个以节点为键、(x,y) 为值的字典。

import networkx as nx
import matplotlib.pyplot as plt

G=nx.read_shp('C:\Users\MyName\MyFolder\TEST.shp') #Read shapefile as graph
pos = {xy: xy for xy in G.nodes()}
nx.draw_networkx_nodes(G,pos,node_size=100,node_color='r')
nx.draw_networkx_edges(G,pos,edge_color='k')
plt.xlim(450000, 470000)
plt.ylim(430000, 450000)

NetworkX：从 shapefile 向图形添加边

NetworkX: add edges to a graph from a shapefile

python

networkx

shapefile

edges