我只是拿起NetworkX并尝试学习如何将它与Shapefile一起使用。
现在我有一个带有道路网络的.shp我希望在带有NetworkX的图表中表示,这样我就能找到2个GPS点之间的最短路径。我尝试使用this,但问题是当我运行write_shp()函数时,我会丢失边缘,因为DiGraph在同一个两个节点之间不允许多个边缘。下图中的箭头显示了使用DiGraph丢失边缘的示例。
所以我想知道是否有任何方法可以创建一个MultiDiGraph,所以我不会丢失任何边缘,或者如果有任何方法可以使用它。我想也许我可以编写一些代码从Shapefile中提取属性并创建MultiDiGraph而不使用NetworkX的read_shp(),但我没有任何使用图形的经验,所以我不确定它是否是有可能。
我非常感谢你能给我的任何帮助或指导,或者如果我错过了任何文件,请告诉我。提前谢谢。
答案 0 :(得分:3)
我可以从您的问题中得到最好的解决方案,以下内容将完成,基本上是从原始的read_shp
命令中复制的。
def read_multi_shp(path):
"""
copied from read_shp, but allowing MultiDiGraph instead.
"""
try:
from osgeo import ogr
except ImportError:
raise ImportError("read_shp requires OGR: http://www.gdal.org/")
net = nx.MultiDiGraph() # <--- here is the main change I made
def getfieldinfo(lyr, feature, flds):
f = feature
return [f.GetField(f.GetFieldIndex(x)) for x in flds]
def addlyr(lyr, fields):
for findex in xrange(lyr.GetFeatureCount()):
f = lyr.GetFeature(findex)
flddata = getfieldinfo(lyr, f, fields)
g = f.geometry()
attributes = dict(zip(fields, flddata))
attributes["ShpName"] = lyr.GetName()
if g.GetGeometryType() == 1: # point
net.add_node((g.GetPoint_2D(0)), attributes)
if g.GetGeometryType() == 2: # linestring
attributes["Wkb"] = g.ExportToWkb()
attributes["Wkt"] = g.ExportToWkt()
attributes["Json"] = g.ExportToJson()
last = g.GetPointCount() - 1
net.add_edge(g.GetPoint_2D(0), g.GetPoint_2D(last), attr_dict=attributes) #<--- also changed this line
if isinstance(path, str):
shp = ogr.Open(path)
lyrcount = shp.GetLayerCount() # multiple layers indicate a directory
for lyrindex in xrange(lyrcount):
lyr = shp.GetLayerByIndex(lyrindex)
flds = [x.GetName() for x in lyr.schema]
addlyr(lyr, flds)
return net
我将返回的图表从DiGraph
更改为MultiDigraph
,我不得不更改add_edge
命令,因为MultiDiGraph
版本的语法与{{1}不同}
答案 1 :(得分:0)
如果多线在关节处断开连接,我认为此库python-s2g
(https://github.com/caesar0301/python-s2g)可以为您提供帮助。甚至networkx的Graph对象也在底层使用,这些多路径实际上是由图形数据记录的。
答案 2 :(得分:-1)
我在这里实现了一个解决方案:https://gitlab.com/njacadieux/upstream_downstream_shortests_path_dijkstra
我使用GeoPandas而不是networkx.readwrite.nx_shp.read_shp读取shapefile。构建图形时,我检查平行边。如果找到而不是像networkx.readwrite.nx_shp.read_shp函数那样跳过它们,我将平行边分成相等长度的两个边,然后将用户长度除以2。必须提供用户长度变量字段名称。