Matplotlib 错误:X 和 Y 必须具有相同的第一维和 lambda 函数

Question

我对python很陌生，尤其是对lambda函数很陌生。我尝试绘制我在下面定义的函数，但我得到

值错误：x 和 y 必须具有相同的第一维，但具有形状 (100，( 和 (1，(

从我所看到的函数绘图来看，我认为错误在于我的函数的定义，即 W 定义得很糟糕。 这是代码：

import numpy as np
import matplotlib.pyplot as plt
def W(t):
s=lambda t: t
for k in range(5):
s=lambda t, y=s: y(t)+k
return s
t=np.linspace(0,1,100)
plt.plot(t,W(t))

Answer 1

如果我将我的代码更改为这样，它可以工作：

import numpy as np
import matplotlib.pyplot as plt
def W(t):
s=lambda x: x
for k in range(5):
s=lambda x, y=s: y(x)+k
return s(t)
t=np.linspace(0,1,100)
plt.plot(t,W(t))