如何用networkx绘制平面图?

Sen*_*hoj 7 matplotlib networkx python-3.x

我正在尝试使用 python 包“matplotlib”和“networkx”绘制有向图的平面肖像。

我尝试使用“networkx.planar_layout”作为图中的节点位置,但不喜欢结果。

在以下示例中,“图”是(平面)有向图。字典“图”的键是节点。键的值是一个列表,其中包含该节点的所有邻居:

import networkx as nx
import matplotlib.pyplot as plt

graph = {'s1': ['v', 't1','w'],
     's2': ['t1','s1'],
     's3': ['v','w'],
     's4': ['x','y'],
     'x': ['v','w'],
     'v': ['t1', 'w'],
     'w': ['y','t1','t2'],
     'y': ['v','t1','t2'],
     't1': [],
     't2': []
     }

def main(G):
    fig = plt.figure()
    fig.show()

    graph = nx.DiGraph()

    for v in G.keys():
        graph.add_node(v)

    for delta in G.items():
        for w in delta[1]:
            graph.add_edge(delta[0],w)

    posit = nx.planar_layout(G)

    nx.draw(graph, posit , with_labels = True)
    fig.canvas.draw()


main(graph)
Run Code Online (Sandbox Code Playgroud)

我得到的图像:在此输入图像描述

我不喜欢的是节点的排列方式会导致边缘“堆叠”。例如,不可能从图中看出边缘 (s2,t1) 真正结束的位置,因为图像的这部分中的边缘全部重叠(我什至认为这不符合平面的定义)我的图表的描绘,这很奇怪,因为我使用的布局称为“planar_layout”,而图表实际上是平面的)。

有更好的方法来绘制这个吗?

Sen*_*hoj 6

我找到了一个可以提供帮助的函数:我使用“nx.draw_planar”而不是“nx.draw”,如下所示:

def main(G):
    fig = plt.figure()
    fig.show()

    graph = nx.DiGraph()

    for v in G.keys():
        graph.add_node(v)

    for delta in G.items():
        for w in delta[1]:
            graph.add_edge(delta[0],w)

    #posit = nx.shell_layout(G) #ISN'T NEEDED ANYMORE

    nx.draw_planar(graph,with_labels = True, alpha=0.8) #NEW FUNCTION
    fig.canvas.draw()


    main(graph)
Run Code Online (Sandbox Code Playgroud)

我得到以下结果: 结果

这个解决方案的问题是我无法像以前的版本一样使用“posit”保存节点位置。不过,我想稍后在程序中使用它们。有谁知道如何在不使用 networkx 布局的情况下获得它们?