使用 matplotlib imshow 在 for 循环中显示 numpy 数组

Question

我有一个 numpy 数组，其元素在 for 循环中更新：

a = np.array([[1,2,3],[4,5,6],[7,8,9]])

for t in range(0,10):
    imshow(a)

    for i in range(0,a.shape[0]):
        for j in range(0,a.shape[1]):
            a[i][j] += 1

我想在每次迭代时显示数组，但是 imshow() 不起作用，它只是在循环终止后显示图像。

ps。我正在使用 Ipython 笔记本

我在网上找到了不同的东西，但它们都不能在我的电脑上运行（例如我尝试使用 matplotlib 的动画模块）

奇怪的是，如果我尝试使用标准 python 提示符执行此示例 (http://matplotlib.org/examples/animation/dynamic_image2.html)，一切正常，而在 Ipython 笔记本上则无法正常工作。谁能解释一下为什么？

注意事项：

也许我过于简化了我的代码；

我正在研究森林火灾模型，数组是一个网格，其中 0 = 空站点、1 = 树、2 = 火灾。

在每个时间步（迭代）：

1. 一棵树被丢弃在一个随机选择的地点，如果该地点空闲，则种植这棵树
2. 一棵树点燃的概率为 f

我想使用颜色图来显示数组，以可视化我的模型的演变

Answer 1

imshow(a) 会将数组 a 的值绘制为像素值，但不会显示该图。每次for循环迭代后查看图片，需要添加show()。

这应该做你想做的：

from matplotlib.pyplot import imshow, show    

a = np.array([[1,2,3],[4,5,6],[7,8,9]])

for t in range(0,10):
    imshow(a)
    show()

    for i in range(0,a.shape[0]):
        for j in range(0,a.shape[1]):
            a[i][j] += 1