2016-08-25 59 views
-1

基本上我嘗試與numpy的陣列的布爾表達式,例如是這樣的:與numpy的數組布爾表達式

import numpy as np 
a = np.array([1,0,1]) 
b = np.array([0,1,1]) 
c = np.array([1,0,1]) 

if (a ==b) or (a==c): 
    d = [2,5,5] 
else : d = [1,5,5] 
print d 

這將返回一個錯誤

ValueError異常:陣列的與真值不止一個元素是不明確的。使用a.any()或a.all()

什麼是正確的方式來編碼?

回答

0

的條件應該是:

if (a==b).all() or (a==c).all(): 

看到這個question