具有多個元素的數組的真值是不明確的。在python

Question

使用a.any（）或a.all（）假設我有這樣的功能：

a = np.array([-2,1,-5,2]) 
if a <0: 
    a += 10 
print (a) 

爲什麼我收到此錯誤。

如果我只傳遞一個值，那麼它是okey。如何解決這個問題呢？

謝謝你提前。

Answer 1

您可以通過索引用布爾陣列（a < 0）簡單的增加：

a[a < 0] += 10 

Answer 2

numpy做元素明智的比較和添加，並矢量化。的if-else在numpy直接翻譯爲np.where()：

import numpy as np 
a = np.where(a < 0, a + 10, a) 
# array([8, 1, 5, 2])