使用matplotlib可视化一组数组

时间:2018-05-31 21:25:45

标签: python animation matplotlib

我想使用matplotlib超时地显示一组数组(中间有一些暂停)。到目前为止我所拥有的是一个阵列的可视化,但我不知道如何使它像动画一样。这是我到目前为止的代码,它创建了一个数组列表,并成功地显示了列表中的第一个数组。

import matplotlib.pyplot as plt
import numpy as np
from PIL import Image
plt.style.use('classic')

a = []
for i in xrange(5):
    temp = np.zeros((5,5))
    temp[i, i] = 10
    a.append(temp)

fig, ax = plt.subplots(5, 5, figsize=(5, 5))
fig.subplots_adjust(hspace=0, wspace=0)

for i in range(5):
    for j in range(5):
        ax[i, j].xaxis.set_major_locator(plt.NullLocator())
        ax[i, j].yaxis.set_major_locator(plt.NullLocator())
        if a[0][i,j] == 10:
            ax[i, j].imshow(Image.open('A.png'), cmap="bone")
        else:
            ax[i, j].imshow(Image.open('B.png'), cmap="bone")

plt.show()

如何将所有列表数组可视化为动画?

1 个答案:

答案 0 :(得分:1)

您需要导入动画模块并定义一个更改框架的功能,以显示您捕获的每个图像。

from matplotlib.animation import FuncAnimation

    def update(i):
        label = 'timestep {0}'.format(i)
        //Insert the data the frame here Eg:ax.imgshow(Image.open('A.png'), cmap="bone")
        ax.set_xlabel(label)
        return line, ax

    FuncAnimation(fig, update, frames=np.arange(0, 10), interval=200)

    plt.show()

您可以按照这个方便的指南进行操作:

https://eli.thegreenplace.net/2016/drawing-animated-gifs-with-matplotlib/