一、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用于比較兩個數組是否相等,且具有一定的誤差容忍度。