Numpy中判斷兩個數組是否相等

 

 

#對於二維的數組，數組形狀相同時才可以判斷，如一個(2,3)的數組和一個(2,2)的數組判斷時會報錯
a=np.array([[1,2,3],[4,5,6]])
b=np.array([[1,2,5],[4,4,5]])

c=(a==b)
d=c.any()  #只要有一個TRUE，就返回TRUE
e=c.all()   #必須都是True，才返回True，否則返回False

print("\n(a==b)的返回結果：\n",c)
print("\n使用any()函數返回的結果:",d)  
print("使用all()函數返回的結果:",e)

結果；

 

 END

 

 

 

#a是多行一列的數組，b是一行多列的數組，如果b是2行一列的數組，則下面的判斷a==b會報錯
a=np.array([[1],[2],[3],[3],[1]]) #(5,1)的數組
b=np.array([1,3])   #(1,2)的數組

c=(a==b)
d=c.any()  #只要有一個TRUE，就返回TRUE
e=c.all()   #必須都是True，才返回True，否則返回False

print("\n數組a的形式：\n",a)
print("\n數組b的形式：\n",b)
print("\n(a==b)的返回結果：\n",c)
print("\n使用any()函數返回的結果:",d)
print("使用all()函數返回的結果:",e)

結果：

 

 END