python - 有没有办法使矩形形状函数为正值？

Question

我正在尝试为一个十秒的信号做一个矩形函数，其值在1到4秒之间为1，其余为0，我查看了其他问题，但它们似乎只涵盖了重复脉冲，而我只是想要这个单脉冲。我已经尝试过下面的代码，但由于我对编程非常陌生，我似乎无法让它工作。我也看到了这个问题，但由于它只给出了绝对值，所以它对我不起作用 rectangular pulse train in python

y=np.zeros(10)
def rect(x):
    x = np.linspace(0, 10, 100)
    if 1<=x<=4:
        y=1
    else:
        y=0
    return rect(x)
f1=rect(y)
plt.plot(y,f1)

Answer 1

有两种方法可以实现:使用函数式方法的长距离方法和向量化的短方法。我介绍了两者:

Longway:在 for 循环中调用该函数并将 y 的值附加到列表中。

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(0, 10, 100)

def rect(i):
    if 1<=i<=4:
        y=1
    else:
        y=0
    return y

f1 = []
for i in x:
    f1.append(rect(i))

plt.plot(x,f1)
plt.show()

快捷方式:创建一个条件掩码并将其应用于 y 数组，根据条件用 0 和 1 填充它。

x = np.linspace(0, 10, 100)
mask = (x>=1) & (x <=4)

y = np.where(mask, 1, 0)
plt.plot(x,y)
plt.show()