Networkx + Matplotlib 动画的问题 - 'NoneType' 对象没有属性 'set_visible'

Question

我在运行以下 DFS 动画时遇到了一些问题。 我相信这可能是因为没有背景画布，但我不确定如何解决这个问题，因为所有其他类似的在线实现都使用 plt.plot 而不是 nx.draw 来保存要显示的图像。

有人可以提供任何指导吗？

import networkx as nx
import matplotlib.pyplot as plt
from matplotlib.animation import ArtistAnimation

fig = plt.figure()
ax = plt.gca()
colors = [0]*len(g)
cmap = plt.get_cmap("autumn")

g = nx.random_tree(20)
pos = nx.fruchterman_reingold_layout(g, k=0.1)


ims = [[nx.draw_networkx(g, pos, node_color = colors, cmap = cmap, ax = ax, vmin=0.0, vmax=1.0)]]

artists = [(nx.draw_networkx(g, pos, node_color = colors, cmap = cmap, ax = ax, vmin=0.0, vmax=1.0),)]

stack = [0] 

while stack:
    node = stack.pop()
    
    if colors[node]: 
        continue
        
    colors[node] = 0.8
    stack += list(g[node].keys())
    
    img = nx.draw_networkx(g, pos, node_color=colors, cmap=cmap, ax=ax, vmin=0.0, vmax=1.0)
    ims += [img]
    
anim = ArtistAnimation(fig, ims, blit = True)
# plt.show()

Answer 1

您的代码的问题是nx.draw_networkx() 不返回任何内容，而使用FuncAnimation 方法总是更容易。首先，您需要创建一个颜色生成器，以使动画函数切换到每次调用设置的下一个颜色。然后使用FuncAnimation 你可以为你的帧（情节）设置动画：

import networkx as nx
import matplotlib.pyplot as plt
import matplotlib
import matplotlib.animation as animation

matplotlib.use('TkAgg')
plt.ion()

g = nx.random_tree(20)
colors = [0] * len(g)
cmap = plt.get_cmap('autumn')
pos = nx.fruchterman_reingold_layout(g, k=0.1)

# here you make the generator
def change_colors():
    stack = [0]
    
    yield colors

    while stack:
        node = stack.pop()
        if colors[node]:
            continue

        colors[node] = 0.8
        stack += list(g[node].keys())

        yield colors

# instantiate your generator
color_gen = change_colors()


def update_frame(n):
    # clear the plot
    plt.cla()
    # here switch to the next colors
    colors = next(color_gen)
    # then draw
    nx.draw(g, pos, with_labels=True, node_color=colors, cmap=cmap, vmin=0.0, vmax=1.0)


ani = animation.FuncAnimation(plt.gcf(), update_frame, repeat=False, interval=1000)

plt.ioff()
plt.show()

这将为您提供： 您可以通过将plt.show() 替换为ani.save('anim.gif', writer='imagemagick') 来将其保存为gif 文件。