使用NumPy计算向量中的值

Question

我使用NumPy。

我用NumPy和其他带有数值的变量定义了一个向量x。

我将在这个矢量y需要返回相同长度的矢量作为y但x值y[i]从根据相应x[i]不同的公式计算的。

我与NumPy的可以做一些智能或我通过矢量x以及用于x每个元件进行迭代确定是否x[i]是除了大于特定值更大或更小，并确定要使用的特定元件，其式？

我想我可以做类似

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

Answer 1

退房np.wherehttp://docs.scipy.org/doc/numpy/reference/generated/numpy.where.html。

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