python - 在 Python 中绘制 Networkx 图

Question

我最近开始使用 python 中的 networkx 库来生成和可视化图表。我从一个简单的代码(由 4 个节点组成)开始，如图所示

import networkx as nx
import matplotlib.pyplot as plt
G = nx.Graph()
G.add_edges_from([(1 ,2) , (2 ,3) , (1 ,3) , (1 ,4) ])
nx.draw(G)
plt.show()

当我连续两次运行代码时，相同代码的输出如图所示(绘图的方向是随机的)

是否可以生成具有相同/固定方向的绘图输出？

Answer 1

默认情况下使用 spring 布局来绘制图形。每次绘制时都可以有一个新的方向。

还有其他可用的布局。

使用例如nx.draw_circular 或 nx.draw_spectral 将为您提供始终相同的布局。

您还可以使用字典定义节点的位置，它将节点编号映射到一个位置。

import networkx as nx
import matplotlib.pyplot as plt
G = nx.Graph()
G.add_edges_from([(1 ,2) , (2 ,3) , (1 ,3) , (1 ,4) ])
pos = { 1: (20, 30), 2: (40, 30), 3: (30, 10),4: (0, 40)} 

nx.draw_networkx(G, pos=pos)

plt.show()