使用NumPy计算向量中的值

时间:2015-04-26 15:19:30

标签: python numpy vector scipy

我使用NumPy。

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

我将返回与y长度相同的向量x,但此向量y[i]中的值y需要根据相应的{不同公式计算{1}}。

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

我想我可以做点像

x[i]

1 个答案:

答案 0 :(得分:6)

结帐np.where http://docs.scipy.org/doc/numpy/reference/generated/numpy.where.html

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