python - 使用 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)