Numpy 动画2D热图

我有一个函数

u（x，y，z，t）

，我想用色条（u_min，u_max）绘制

（x，y）

动画热图，其中

t

是时间动画的参数。我已经分析了这些例子，但看起来很复杂。

z

参数将保持不变。定义的

u（x，y，z，t）

是方程

我做的

import matplotlib.pyplot as plt
import numpy as np

def u(x, y, z, t):

   alpha = 9*10**-6
   a = 0.01
   z = a/2


   for i in range(1,k):

       sin1 = np.sin((i*np.pi*x)/a)

       sin2 = np.sin((i*np.pi*y)/a)

       sin3 = np.sin((i*np.pi*z)/a)

       e = np.exp(-((alpha**2)*(np,pi**2)*((n**2)*(m**2)*(l**2))*t/a**2)

       a_n = 80*(2/np.pi)**3*(1/n*m*l)*(1-(-1)**n)*(1-(-1)**m)*(1-(-1)**l)

       u += 20 + a_n*sin1*sin2*sin3*e

    return u

data = u

fig, ax = plt.subplots()

for j in range(len(data)):
    ax.cla()
    ax.imshow(data[j])
    ax.set_title("frame {}".format(i))
    plt.pause(0.1)

---

谢谢您的帮助。

可能会解决您的问题。参考。我在帖子中解释了更多。这不是我们需要让函数工作的东西吗？让我们尝试很多东西。

ax.imshow（u（x，y，z，t，j））