NetworkX：从 shapefile 创建批处理图

debugcn 发表于 Dev

法式咖啡

假设您有以下包含多个 shapefile 的目录：

MyFolder
|
|---S1.shp
|---S2.shp
|---S3.shp
|---
|---
|---S50.shp

并且您想为这些中的每一个创建一个图，并称它们为相同的（例如，与之关联的图S1.shp称为S1）。

我正在尝试在 for 循环中执行此操作：

import networkx as nx
import os

myfolder=r'C:\Users\MyName\MyFolder'
os.chdir(myfolder) 
for root, dirs, filenames in os.walk(myfolder):
    for f in filenames:
        if f.endswith('.shp') and f[0]=='S': #Consider only the files with an "S" in their name
            G=nx.read_shp(f)
            pos={k: v for k,v in enumerate(G.nodes())}
            pos_x = {k:(round(a,4),round(b,4)) for k,(a,b) in pos.items()}
            graph_name=f.split('.', 1)[0]
            GG=nx.Graph(name=graph_name)  
            GG.add_nodes_from(pos_x.keys())

现在这GG=nx.Graph(name=graph_name)条线没有做我想要的，因为该图仍然被称为GG，而不是S1诸如此类。正确的图形名称存储在中GG.name，但我无法通过调用其 GG.name 来检索图形。

有没有办法可以为图形指定一个名称，即文件名，而没有扩展名？

忘了它

您可以制作GG一个将文件名映射到图形的字典：

import networkx as nx
import os

myfolder = r'C:\Users\MyName\MyFolder'
os.chdir(myfolder)
GG = dict()
for root, dirs, filenames in os.walk(myfolder):
    for f in filenames:
        if f.endswith('.shp') and f.startswith('S'):
            G = nx.read_shp(f)
            pos_x = {k: (round(a, 4), round(b, 4))
                     for k, (a, b) in enumerate(G.nodes())}
            graph_name, graph_ext = os.path.splitext(f)
            GG[graph_name] = nx.Graph(name=graph_name)
            GG[graph_name].add_nodes_from(pos_x.keys())

然后S1.shp可以通过访问与关联的图GG['S1']。

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。