利用numpy生成各种波

一，生成矩形波

矩形波的傅里叶级数

x=np.linspace(-np.pi,np.pi,201)
k=np.arange(1,99)
k=2*k-1
y=np.zeros_like(x)
for i in range(len(x)):
    y[i]=(4/np.pi)*np.sum(np.sin(k*x[i])/k)
# print(t)
# print(f)
plt.plot(x,y)
plt.show()

二，生成锯齿波和三角波

锯齿波的傅里叶级数

x = np.linspace(-np.pi, np.pi, 201)
k = np.arange(1, 99)
y = np.zeros_like(x)
for i in range(len(x)):
    y[i] = -(2 / np.pi) * np.sum(np.sin(2*np.pi * x[i]*k) / k)
# print(t)
# print(f)
plt.plot(x, y)
# plt.show()

plt.plot(x, abs(y))
plt.show()