python numpy 三角函数 合成方波

python numpy 三角函数 合成方波

"""
    三角函数 合成方波
"""
import numpy as np
import matplotlib.pyplot as mp

# 准备x值与y值
x = np.linspace(0, np.pi * 4, 1000)
y1 = 4 * np.pi * np.sin(x)
y2 = 4 / 3 * np.pi * np.sin(3 * x)
y3 = 4 / 5 * np.pi * np.sin(5 * x)

# 叠加1000条曲线 组成方波
n = 1000
y = np.zeros(n)
for i in range(1, n + 1):
    y += 4 / (2 * i - 1) * np.pi * np.sin((2 * i - 1) * x)

# 绘制图片
mp.grid(linestyle=":")
mp.plot(x, y1, label="y1", alpha=0.2)
mp.plot(x, y2, label="y2", alpha=0.2)
mp.plot(x, y3, label="y3", alpha=0.2)
mp.plot(x, y, label="y")

mp.legend()
mp.show()