Python证明方波的谐波分量

最近在学习python各种库之间的运用。
比如matplotlib的练习，用Python来证明方波的谐波分量。
代码如下：
def square_sigma_sin(x, n):
“”“将n个三角波函数合成为方波的近似函数”""
sum1 = 0
for a in range(1, 2*n, 2):
sum1 += 1 / a * math.sin(a * x)
return sum1
N = 200 # N个采样点
input_value = [4 * math.pi * x / N for x in range(1, N + 1)] # 输入列表，两个周期，将4pi分为N份
output_value = [square_sigma_sin(x, 9) for x in input_value] # 输出列表，将输入带入到近似函数

plt.title(“Square Function”, fontsize=24)
plt.axis([0, 4 * math.pi, -2, 2])
plt.xlabel(“input”, fontsize=16)
plt.ylabel(“output”, fontsize=16)
plt.tick_params(axis=“both”, labelsize=16)
plt.scatter(input_value, output_value, c=output_value, cmap=plt.cm.Set2, s=20)
plt.savefig(‘squares_plot.png’, bbox_inches=‘tight’)
plt.show()