Python

Question

我对使用 Networkx 和 pygraphviz 比较陌生（请多多包涵...）我有一个图表，我使用 Networkx 和 graphviz_layout.

进行了可视化

这一行之后：

pos=graphviz_layout(G, prog='dot')

我检索了图表中的 x、y 坐标列表，如下所示：

#coordinates of the nodes
node_x = []
node_y = []
for(node,(x,y)) in pos.items():
    node_x.append(x)
    node_y.append(-y)

有没有办法检索图中边的坐标并将其附加到列表中？例如

#coordinates of the edges:
edge_x = []
edge_y = []
#how do i get the edge coordinates set by graphviz_layout here?
     edge_x.append(x0)
     edge_x.append(x1)
     edge_x.append(None)
     edge_y.append(y0)
     edge_y.append(y1)
     edge_y.append(None)

如有任何帮助，我们将不胜感激！提前致谢！

Answer 1

由于我没有安装 graphviz_layout，假设它与其他布局函数的工作原理类似，我创建了一个具有标准布局函数的测试。可以通过边循环得到坐标

以下代码使用list comprehension使代码更紧凑。最后 plt.plot(edge_x, edge_y) 用于可视化创建的列表。

import networkx as nx
import matplotlib.pyplot as plt

G = nx.complete_graph(20)
pos = nx.circular_layout(G)
node_x = [x for x, y in pos.values()]
node_y = [-y for x, y in pos.values()]

edge_x = [x for n0, n1 in G.edges for x in (pos[n0][0], pos[n1][0], None)]
edge_y = [y for n0, n1 in G.edges for y in (-pos[n0][1], -pos[n1][1], None)]

plt.plot(edge_x, edge_y, color='purple', lw=0.5)
plt.scatter(node_x, node_y, color='crimson', s=50, zorder=3)
plt.gca().set_aspect('equal')
plt.axis('off')
plt.show()

Python - 从 networkx-graphviz_layout 获取边缘坐标

Python - Get edge coordinates from networkx-graphviz_layout

graphviz

networkx

pygraphviz