グラデーションを用いて数値計算の結果を可視化したい

数値計算で、1列30個のセルに0以上1以下の値を持たせて、その値に応じて色を付けたいと思っています。
0に近いほど白に近く、1に近いほど黒に近いグラデーションで表したいと思っています。
そしてアニメーションにしてステップごとのセルの色を表示したいと思っています。

そこでcmapを使おうとしているのですが、matplotlibでまず黒色を背景とする画面を作り、その後cmapで値によって濃さ（黒さ）が変わるグラデーションを作り、その後ステップごとのアニメーションにしようと思っていますが、それをどうコードに置き換えればいいのか分かりません。

動画の前に、matplotlibについて書かれているサイトなどを参考にして2ステップ目の画像を作ろうとして以下のコードを書いて、無事画像ファイルまでは出せたのですが、その画像ファイルが予期せぬことに、2ステップ目の画像ではなく、cmapのグラデーションも入っていなければ、横のセルの連続を作りたいはずが、縦向きの列になっていて、さらにセルやセルの色が反映されていません。どこがおかしいのでしょうか？

execCell(A,i1)の値が正しく出ることは、print関数と手計算で確かめて、確かめたどのセル、ステップでも値は同じなのでdef imageCell():の前までは問題ないと思います。

コード

python3
1import numpy as np
2import time
3import random
4import math
5import matplotlib.cm as cm
6import matplotlib.pyplot as plt
7
8def array2(N1,N2): 
9    return [[0 for j in range(N2)]for k in range(N1)]
10def array3(N1,N2,N3): 
11    return [array2(N2,N3)for k in range(N1)]
12
13def initdt():
14    A=array3(1001,31,2)
15    A[0][15][1]=0.80
16    return A
17
18def execCell(A,i1):
19    i2 = 1 + i1
20    for i in range(1,30):
21        A[i2][i][1] = 2 * A[i1][i][1] * (1 - A[i1][i][1]) + A[i1][i-1][1] * (1 - A[i1][i-1][1]) + A[i1][i+1][1] * (1 - A[i1][i+1][1])
22
23def imageCell(A,k):
24    plt.figure()
25    for k in range(0,1000):
26        for i in range(1,30):
27            execCell(A,k)
28            plt.imshow(array2(i,1), cmap=cm.Greys)
29    plt.show(array2(i,1))
30
31A=initdt()
32imageCell(A,2)