Вычисление значений в векторе с 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)