Заменить значение в массиве с условием

Question

У меня есть эта упрощенная программа для замены значений в массиве, которые удовлетворяют условиям:

formula1=2*2 
formula2=5*2 
formula3=4*4 

array = np.random.rand(2,4,10) 
for n,i in enumerate(array): 
    if i>0.5: #find value in array with this condition 
     formula = formula1 
     array[n] = array[n]*formula #replace the found value with this value 
    elif i <0.1: 
     formula = formula2 
     array[n] = array[n]*formula 
    else: 
     formula = formula3 
     array[n] = array[n]*formula 
print array  

Это привело к сообщению об ошибке: "Значение истинности массива с более чем один элемент неоднозначный. Используйте a.any() или a.all() '. Любое предложение?

Answer 1

Чтобы перечислить массив, вы можете использовать np.ndenumerate (documentation here):

for n, i in np.ndenumerate(array): 
    ...