np.equal詳解

一、np.equal用法

numpy庫中的np.equal函數用於對比兩個數組相同位置的元素是否相等，返回一個布爾值的數組（True表示相等，False表示不相等）。該函數的參數有兩個：x1和x2。其中x1和x2可以是數組或標量。當x1和x2具有不同的類型時，嘗試將它們轉換為相同的類型。

import numpy as np

a = np.array([1, 2, 3, 4])
b = np.array([1, 2, 5, 6])

print(np.equal(a,b)) #[ True  True False False]

上述例子中，數組a和數組b都有四個元素，np.equal函數對比數組a和數組b相同位置的元素，返回一個新的包含四個元素的布爾值數組。

二、np.equal()返回值

np.equal()函數返回一個與第一個參數相同的形狀的布爾值數組。np.equal()函數的返回值是由兩個輸入數組逐位置進行比較後得到的結果。

import numpy as np

a = np.array([1, 2, 3, 4])
b = np.array([1, 2, 5, 6])
c = np.array([1, 2, 3, 4])

print(np.equal(a,b)) #[ True  True False False]
print(np.equal(a,c)) #[ True  True  True  True]
print(np.equal(b,c)) #[ True  True False False]

上述代碼中，np.equal函數比較a和b數組元素的值，並將比較結果「逐位置」存儲到一個新的數組中返回。對於a和c，它們的元素值都一樣，所以返回的數組的每個位置都是True。對於b和c，它們的元素值不完全相同，所以返回的數組中只有前兩個位置是True。

三、np.equal函數相關用法

1. np.array_equal函數的用法

與np.equal不同，np.array_equal用於比較兩個數組是否相等。

import numpy as np

a = np.array([1, 2, 3, 4])
b = np.array([1, 2, 5, 6])
c = np.array([1, 2, 3, 4])

print(np.array_equal(a,b)) #False
print(np.array_equal(a,c)) #True

上述代碼中，我們用np.array_equal比較數組a和數組b的值，返回False；比較數組a和數組c的值，返回True。

2. np.allclose函數的用法

np.allclose函數用於比較兩個數組是否相等，這兩個數組可以有一定的誤差容忍度。np.allclose的用法為np.allclose(a,b,atol=1e-8,rtol=1e-5)。其中a、b是兩個數組，atol是絕對誤差容忍度（默認值是1e-8），rtol是相對誤差容忍度（默認值是1e-5）

import numpy as np

a = np.array([1, 2, 3, 4])
b = np.array([1.001, 2.001, 3.001, 4.001])

print(np.allclose(a,b,atol=0.01,rtol=0.01)) #True
print(np.allclose(a,b,atol=0.0001,rtol=0.01)) #False

上述代碼中，我們認為a和b的誤差容忍度分別是0.01和0.0001，第一次比較得到結果為True，第二次比較得到結果為False。

四、總結

np.equal函數是numpy庫中用於對比兩個數組相同位置的元素是否相等的函數，返回一個布爾值的數組。同時，np.array_equal和np.allclose也是對比兩個數組是否相等的方法。np.array_equal用於比較兩個數組是否相等，而np.allclose用於比較兩個數組是否相等，且具有一定的誤差容忍度。