如何用python绘制奇偶方波

import numpy as np
import matplotlib.pyplot as plt

# 设置方波的参数
amplitude = 1  # 方波的幅度
frequency = 1  # 方波的频率
duration = 2   # 方波的持续时间

# 生成时间序列
t = np.linspace(0, duration, num=1000)

# 生成奇偶方波信号
odd_wave = amplitude * np.sign(np.sin(2 * np.pi * frequency * t))
even_wave = amplitude * np.sign(np.cos(2 * np.pi * frequency * t))

# 绘制奇偶方波
plt.plot(t, odd_wave, label='Odd Wave')
plt.plot(t, even_wave, label='Even Wave')

# 设置图形属性
plt.xlabel('Time')
plt.ylabel('Amplitude')
plt.title('Odd and Even Square Waves')
plt.legend()

# 显示图形
plt.show()