Как генерировать непериодический квадратный сигнал в scipy

Question

Я хотел бы перевести следующий простой квадрат из функции 0 в 1 в scipy.signal, чтобы использовать операцию свертки с другим scipy.signal

def f(t): return 1 if (np.floor(t) < 1) else 0

Как я могу это сделать?

Редактировать

Извините за путаницу, предполагая, что у меня есть следующее определение функции:
return np.where(np.floor(t) < 1, 1 , 0)

Как я могу написать версию scipy.signal?

Answer 1

NumPy's where может быть тем, что вы ищете:

def f(t):
    return np.where(np.floor(t) < 1, 1 , 0)