python - 用 Python 绘制方波

Question

我目前正在使用 numpy 和 pylot 在 python 中绘制方波图。如何在 T 的多个周期上绘制方波函数？

我目前有:

from scipy import signal
import numpy as np
from scipy.fftpack import fft

#Initialize Parameters
p_0 = 2
A = np.sqrt(1/(2*p_0))
t = [-A,A]


plt.plot(t,[A,A])
plt.show()

这只是给了我一条直线。最终游戏是对方波函数进行傅里叶变换

Answer 1

您可以使用 scipy.signal 中的 square 函数

from scipy import signal
import matplotlib.pyplot as plt
t = np.linspace(0, 1, 500, endpoint=False)
plt.plot(t, signal.square(2 * np.pi * 5 * t),'b')
plt.ylim(-2, 2)
plt.grid()
plt.show()