Python 用NumPy计算向量中的值

我用NumPy

我已经用NumPy定义了一个向量

x

，并用数值定义了其他变量

我将返回与

x

长度相同的向量

y

，但此向量

y

中的值需要根据相应的

x[I]

从不同的公式计算

我可以用NumPy做些聪明的事情吗？或者我必须遍历向量

x

，并为

x

中的每个元素确定

x[I]

是否大于或小于特定值，并确定用于特定元素的公式

我想我可以做一些类似的事情

y[x > a] = 2*x+7
y[x <= a] = 3*x+9
return y

y[x>a]=2*x+7
y[x检查
np.where

y = np.where(x > a, 2 * x + 7, 3 * x + 9)