python - 如何在 python 中绘制两个变量的数学表达式？

Question

我有一个随时间和深度变化的函数，简化为:

def f(z,t):
    return np.exp(-z)*np.sin(t-z)

z = np.linspace(0,3000,num=3001)
t = np.arange(0,40000,4000)

我想在 t 中的每个时间步绘制所有 z 的结果，结果如下:

但是我不知道怎么办。我敢肯定这很简单，但我不习惯在 python 中这样做。

Answer 1

import matplotlib.pyplot as plt
import numpy as np

def f(z,t):
    return np.exp(-z)*np.sin(t-z)

z = np.linspace(0,5,3001)
t = np.arange(0,40000,4000)

for tval in t:
    plt.plot(z, f(z, tval))
plt.show()