Python numpy array_equal用法及代碼示例

本文簡要介紹 python 語言中 numpy.array_equal 的用法。

用法:

numpy.array_equal(a1, a2, equal_nan=False)

如果兩個數組具有相同的形狀和元素，則為 True，否則為 False。

參數：

a1, a2： array_like

輸入數組。

equal_nan： bool

是否比較 NaN 相等。如果 a1 和 a2 的 dtype 是複數，如果給定值的實部或虛部是 nan ，則值將被視為相等。

返回：

b： bool

如果數組相等，則返回 True。

例子：

>>> np.array_equal([1, 2], [1, 2])
True
>>> np.array_equal(np.array([1, 2]), np.array([1, 2]))
True
>>> np.array_equal([1, 2], [1, 2, 3])
False
>>> np.array_equal([1, 2], [1, 4])
False
>>> a = np.array([1, np.nan])
>>> np.array_equal(a, a)
False
>>> np.array_equal(a, a, equal_nan=True)
True

什麽時候equal_nan為 True，則具有 nan 分量的複數值被認為是相等的，如果實數或者虛部為 nan。

>>> a = np.array([1 + 1j])
>>> b = a.copy()
>>> a.real = np.nan
>>> b.imag = np.nan
>>> np.array_equal(a, b, equal_nan=True)
True