NetworkX - 如何从Shapefile创建MultiDiGraph?

时间:2015-06-11 02:55:03

标签: python networkx shapefile digraphs

我只是拿起NetworkX并尝试学习如何将它与Shapefile一起使用。

现在我有一个带有道路网络的.shp我希望在带有NetworkX的图表中表示,这样我就能找到2个GPS点之间的最短路径。我尝试使用this,但问题是当我运行write_shp()函数时,我会丢失边缘,因为DiGraph在同一个两个节点之间不允许多个边缘。下图中的箭头显示了使用DiGraph丢失边缘的示例。

enter image description here

所以我想知道是否有任何方法可以创建一个MultiDiGraph,所以我不会丢失任何边缘,或者如果有任何方法可以使用它。我想也许我可以编写一些代码从Shapefile中提取属性并创建MultiDiGraph而不使用NetworkX的read_shp(),但我没有任何使用图形的经验,所以我不确定它是否是有可能。

我非常感谢你能给我的任何帮助或指导,或者如果我错过了任何文件,请告诉我。提前谢谢。

3 个答案:

答案 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-s2ghttps://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。必须提供用户长度变量字段名称。