Python：如何可视化网络的最小生成树？

Question

我有一个由NxN个节点组成的常规网格网络。我可以绘制它，但我也想绘制它的minimum spanning tree，这是一个“包含所有顶点并且是树的子图”。

这就是我创建网络的方式：

from __future__ import print_function, division
import numpy
from numpy import *
import networkx as nx
from networkx import *
import matplotlib.pyplot as plt

N=30
G=nx.grid_2d_graph(N,N)
pos = dict( (n, n) for n in G.nodes() )
labels = dict( ((i, j), i + (N-1-j) * N ) for i, j in G.nodes() )
nx.relabel_nodes(G,labels,False)
inds=labels.keys()
vals=labels.values()
inds.sort()
vals.sort()
pos2=dict(zip(vals,inds))
nx.draw_networkx(G, pos=pos2, with_labels=False, node_size = 15)

这就是我计算最小生成树的方式：

T=nx.minimum_spanning_tree(G)

这会生成一个与G类似的图表，区别在于T与G具有相同的节点，并且选择了其边缘。因此，我想绘制T，这就是我所做的：

plt.figure()
nx.draw_networkx(G, pos=pos2, with_labels=False, node_size = 15)
plt.show()

但我只是再次绘制网络。如何修改最后一行以确保绘制最小生成树？

Answer 1

plt.figure()
nx.draw_networkx(G, pos=pos2, with_labels=False, node_size = 15)
plt.show()

应该是

plt.figure()
nx.draw_networkx(T, pos=pos2, with_labels=False, node_size = 15)
plt.show()