グラデーションのアニメーションの作り方

Question

質問内容

時間ごとに各セルの値が0から1の範囲で変化して、それを[0,1]の範囲で値が0で白、そして1に近づくにつれて黒に近づくグラデーションで表そうとしていましたが、以下のコードから先がうまく書けません。

各セルのkステップの値を出せるようなプログラムは書けました。（下のコードに、以下のコードを加えて、各セルの値を出した後に、手計算で値を計算して値が正しいか確かめました。）
A=initdt()
k=0
for k in range(0,1000):
execCell(A,k)

print(A[1][14][1])
その後ここで過去にした質問や参考書を頼りに、セルの値を色で表現しようとしてますが、そこで完全に詰んでいる状態です。この先どう書けばよいのでしょうか？

下のコードはセルの値を出すまでのコードです。

import numpy as np
import time
import random
import math

def array2(N1,N2): 
    return [[0 for j in range(N2)]for k in range(N1)]
def array3(N1,N2,N3): 
    return [array2(N2,N3)for k in range(N1)]

def initdt():
    A=array3(1001,31,2)
    A[0][15][1]=0.80
    return A

def execCell(A,i1):
    i2 = 1 + i1
    for i in range(1,30):
        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])

Answer 1

この問題は解決しました。ありがとうごさいます。

ちなみにコードに関しては
https://teratail.com/questions/163020

こちらの質問のKSwordOfHasteさんが書いてくれましたコードを参照してください。