将LineCollection转换为数组或对Networkx图进行动画处理

Question

将LineCollection转换为数组或对Networkx图进行动画处理

4

我正在尝试使用networkx.draw函数的输出结果(LineCollection)，以便在需要数组的matplotlib.animation中使用。我不想将我的图形保存为png格式，因为会有很多这样的图形。我也不想显示它，但这并不是必要的。

以下是一个简单的代码示例：

import networkx as nx
graph= nx.complete_graph(5) #a simple graph with five nodes
Drawing=nx.draw(graph)

这会输出一个Python集合：

<matplotlib.collections.LineCollection at 0xd47d9d0>

我希望创建一个这样的图纸列表：

artists=[]
artists.append(Drawing)

并将这些图纸用于动画制作：

import matplotlib
fig= plt.figure()  #initial figure, which can be empty
anim=matplotlib.animation.ArtistAnimation(fig, artists,interval=50, repeat_delaty=1000)

然而，我遇到了以下 TypeError 错误：

TypeError: 'LineCollection' object is not iterable

所以，我认为“艺术家列表”应该是一组图像，这些图像应该是numpy数组、png图像或一些称为PIL的东西（我不熟悉），我不知道如何将集合转换为其中之一而不保存图像为png或任何其他格式。

实际上，这就是我想要做的事情：一个动态动画，当我尝试使用im = plt.imshow(f(x, y))与我拥有的绘画之一时，会出现此错误：

TypeError: Image data can not convert to float

我希望我表达清晰了，这是我第一次使用动画和绘图工具。有没有人能提供一个解决方案?

- potato

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- ChrisE · Accepted Answer

这里有一个动态动画（如果您想以这种方式查看，它可以在iPython笔记本中工作）。本质上，您想使用draw_networkx并为每个帧提供要绘制的项目。为了防止每次调用该函数时位置发生变化，您需要重复使用相同的位置（如下所示pos）。

%pylab inline  #ignore out of ipython notebook
from IPython.display import clear_output #ignore out of ipython notebook

import networkx as nx
graph= nx.complete_graph(5) #a simple graph with five nodes

f, ax = plt.subplots()

pos=nx.spring_layout(graph)

for i in range(5):
    nx.draw_networkx(graph, {j:pos[j] for j in range(i+1)}, ax=ax, nodelist=graph.nodes()[0:i+1], 
                     edgelist=graph.edges()[0:i], with_labels=False)
    ax.axis([-2,2,-2,2]) #can set this by finding max/mins

    time.sleep(0.05)
    clear_output(True) #for iPython notebook
    display(f)
    ax.cla() # turn this off if you'd like to "build up" plots

plt.close()