如何在matplotlib中的动画期间删除轮廓线

Question

我将 Python 3.6.3 与 matplotlib 2.1.0 一起使用，我想要实现的是能够删除一些在动画之前使用 contour() 函数绘制的轮廓。 等高线图需要在动画开始时可见，并且只有在用户按下某个键后才能删除。困难在于我需要对动画使用 blitting。然而，blitting 会恢复原始画布，因此即使移除轮廓，它仍然在 blitted 背景中保持可见。

这是显示问题的代码：

import matplotlib.pyplot as plt
import matplotlib.animation as animation
from numpy import mgrid, sin, cos, pi

def keypress(event):
    if event.key == "delete":
        for c in cs.collections: c.remove() # this doesn't work
        cs.set_alpha(0.0) # this doesn't work either

phi = pi/2

def animate(i):
    global phi
    line.set_data([[0.0, sin(phi)], [0.0, cos(phi)]])
    phi += 0.01
    return line,

x,p = mgrid[-2:2:100*1j,-2:2:100*1j]
fig,ax = plt.subplots(1, 1, figsize=(19.2,10.8), dpi=100)
ax.set_xlim([-2,2])
ax.set_ylim([-2,2])
line, = ax.plot([], [], 'o-', lw=2, color='b')
cs = ax.contour(x, p, x**2 + p**2, levels=2, linewidths=0.8, colors='r')
fig.canvas.mpl_connect('key_press_event', keypress)
ani = animation.FuncAnimation(fig, animate, blit=True, interval=0, frames=2000)
plt.show()

如果无法删除轮廓，那么至少我应该能够使其不可见或透明，但我也找不到这样做的方法。请帮忙！ :)

Answer 1

在使用blit=True 制作动画期间，很难更改画布中的任何内容（blitting 的全部意义在于背景不会改变）。在不使用 blit 的情况下，删除线条或将其设置为不可见应该没有任何问题，blitting 将始终恢复原始画布背景，即使某些艺术家已从中删除。

因此，我们需要做的是在艺术家移除后中断动画，重新绘制画布，拍摄新的背景快照并使用新背景重新启动动画。虽然这本身会相当麻烦，但可能的解决方案是模拟画布的调整大小事件，这将准确触发所描述的过程。这是在以下代码中完成的。

（请注意，这与Qt 后端一起正常工作，而使用Tk 会出现一些问题，至少在matplotlib 2.1 版中。）

import matplotlib.pyplot as plt
import matplotlib.animation as animation
from numpy import mgrid, sin, cos, pi

phi = pi/2.

def keypress(event):
    if event.key == "delete":
        try:
            for c in cs.collections: 
                c.remove()
        except ValueError:
            pass
        ani._handle_resize()
        ani._end_redraw(None)


def animate(i):
    global phi
    line.set_data([[0.0, sin(phi)], [0.0, cos(phi)]])
    phi += 0.01
    return line,

x,p = mgrid[-2:2:100*1j,-2:2:100*1j]
fig,ax = plt.subplots(1, 1, figsize=(8,6), dpi=100)
ax.set_xlim([-2,2])
ax.set_ylim([-2,2])
line, = ax.plot([], [], 'o-', lw=2, color='b')
cs = ax.contour(x, p, x**2 + p**2, levels=2, linewidths=2, colors='r')

fig.canvas.mpl_connect('key_press_event', keypress)

ani = animation.FuncAnimation(fig, animate, 
                              blit=True, interval=1, frames=2000)
plt.show()