使用matplotlib绘制带有轮廓的动画

Question

我正在尝试在matplotlib 1.1.0中对轮廓进行动画处理。

参考以下链接： http://www.mail-archive.com/[e-mail]/msg17614.html, http://matplotlib.1069221.n5.nabble.com/Matplotlib-1-1-0-animation-vs-contour-plots-td18703.html 和 http://www.scipy.org/Cookbook/Matplotlib/Gridding_irregularly_spaced_data

from numpy import linspace,exp,vstack
from scipy.interpolate import griddata
import matplotlib.pyplot as plt
import matplotlib.animation as animation
from numpy.random import uniform, seed

def main():
    seed(1234)
    x = uniform(-2,2,100)
    y = uniform(-2,2,100)
    data = vstack((x*exp(-x**2-y**2),0.5*x*exp(-x**2-y**2),0.2*x*exp(-x**2-y**2)))
    xi = linspace(min(x), max(x))
    yi = linspace(min(y), max(y))
    zi = []
    numframes = data.shape[0]
    for ii in range(numframes):
        zi.append(griddata((x, y), data[ii], (xi[None,:], yi[:,None]), method='cubic'))

    fig = plt.figure()
    im = plt.contour(xi, yi, zi[0], 15, linewidths=0.5, colors='k')
    ani = animation.FuncAnimation(fig, update_contour_plot, frames=xrange(numframes), fargs=(zi, im, fig, xi, yi), interval=100)
    plt.colorbar()
    plt.show()

def update_contour_plot(i, data, im, fig, xi, yi):
    for coll in im.collections:
        try:
            plt.gca().collections.remove(coll)
        except ValueError: #Everything is not removed for some reason!                                                                                      
            pass
    im = plt.contour(xi, yi, data[i], 15, linewidths=0.5, colors='k')
    plt.title(str(i))
    return im,

main()

1) 这是最好的方法吗？如果有其他更好的方法，请指出。

2) 在最终输出中，前一帧的等高线仍然可见。我该如何删除它们？参见 except ValueError：

- chaitu

---

ValueError是否携带任何消息？ - tacaswell

没有try: except:读出来是这样的plt.gca().collections.remove(coll) ValueError: list.remove(x): x不在列表中 - chaitu

+1 对于好的示例代码 - tacaswell

Answer 1

您遇到了一个范围问题，您在更新程序中创建的新im没有被传递回来，因此第二次循环时，您尝试再次删除第一层线条，但它正确地抱怨无法删除它们，因为您已经删除了它们。

稍微调整一下您的代码：

def main():
    seed(1234)
    x = uniform(-2,2,100)
    y = uniform(-2,2,100)
    data = vstack((x*exp(-x**2-y**2),0.5*x*exp(-x**2-y**2),0.2*x*exp(-x**2-y**2)))
    xi = linspace(min(x), max(x))
    yi = linspace(min(y), max(y))
    zi = []
    numframes = data.shape[0]
    for ii in range(numframes):
        zi.append(griddata((x, y), data[ii], (xi[None,:], yi[:,None]), method='cubic'))

    fig = plt.figure()
    im = plt.contour(xi, yi, zi[0], 15, linewidths=0.5, colors='k')
    ax = fig.gca()
    ani = animation.FuncAnimation(fig, update_contour_plot, frames=xrange(numframes), fargs=(zi, ax, fig, xi, yi), interval=100)
    plt.colorbar(im)
    plt.show()
    return ani


def update_contour_plot(i, data,  ax, fig, xi, yi):
    ax.cla()
    im = ax.contour(xi, yi, data[i], 15, linewidths=0.5, colors='k')
    plt.title(str(i))
    return im,

另一个选择是将im变成某种可变的形式（比如将其包装在字典中），这样你所做的更改就会从帧传播到帧。

- tacaswell

---

ValueError: list.remove(x)：列表中没有 x。 - chaitu