squarewave

numpy绘制方波

方波函数

$$\sum _{k=1}^{\infty }\frac{4\times sin((2k-1)t)}{(2k-1)\times \pi }$$

import numpy as np
import matplotlib.pyplot as plt

k = 4
# 步长
t = np.linspace(-np.pi, np.pi, 201)
k = np.arange(1, 100)
k = 2*k-1

# 和t形状一样
f = np.zeros_like(t)
for i in range(len(t)):
    f[i] = np.sum(4 * np.sin(k*t[i]) / k*np.pi)

plt.plot(t, f)
plt.show()