scipy.signal.square

scipy.signal.square(t, duty=0.5)

Returns a periodic square-wave waveform with period 2*pi which is +1 from 0 to 2*pi*duty and -1 from 2*pi*duty to 2*pi duty must be in the interval [0,1]