使用NumPy计算向量中的值

Question

我用NumPy.

我已经定义了一个x带有NumPy 的向量和其他带有数值的变量.

我将返回一个y长度相同的向量,x但是y[i]该向量中的值y需要根据相应的不同公式计算x[i].

我可以使用NumPy做一些聪明的事情,或者我是否必须迭代向量x并x确定每个元素是否x[i]大于或小于特定值并确定用于特定元素的公式？

我想我可以做点什么

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

Answer 1

查看np.where http://docs.scipy.org/doc/numpy/reference/generated/numpy.where.html.

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