我使用NumPy。
我已经使用NumPy和其他带数值的变量定义了一个向量x
。
我将返回与y
长度相同的向量x
,但此向量y[i]
中的值y
需要根据相应的{不同公式计算{1}}。
我可以让NumPy做一些聪明的事情,或者我是否必须遍历向量x[i]
并对x
中的每个元素确定x
是否大于或小于特定值并确定用于特定元素的公式?
我想我可以做点像
x[i]
答案 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)