如何使用Python将正弦波转换为方波(0或1)？

import numpy as np
from scipy import signal
import matplotlib.pyplot as plt

frequency = 1  # 正弦波频率
amplitude = 1  # 正弦波振幅
sampling_rate = 1000  # 采样率
duration = 1  # 正弦波持续时间

t = np.linspace(0, duration, int(sampling_rate * duration), endpoint=False)
sin_wave = amplitude * np.sin(2 * np.pi * frequency * t)

square_wave = np.where(sin_wave >= 0, 1, 0)

plt.plot(t, sin_wave, label='Sin Wave')
plt.plot(t, square_wave, label='Square Wave')
plt.xlabel('Time')
plt.ylabel('Amplitude')
plt.legend()
plt.show()