如何比較 NumPy 數組的相等性？綜合指南

比較 NumPy 陣列是否相等：綜合指南

使用 NumPy 陣列時，比較它們是否相等是一項常見任務。然而，簡單地使用相等運算子 (==) 會產生一個指示元素相等的布林數組。要確定數組的整體相等性，需要找到更簡潔的方法。

快速解

最直接的解法是使用 (A== B).all() 語句。如果按元素比較數組 (A==B) 的所有元素都為 True，則此表達式的計算結果為 True，表示兩個數組具有相同的元素。

<code class="python">import numpy as np

A = np.array([1, 2, 3])
B = np.array([1, 2, 3])

# Element-wise comparison
are_equal = (A == B).all()

print(are_equal)  # Output: True</code>

特殊情況的注意事項

需要注意的是，這種方法在某些情況下可能會表現出意外的行為：

• 如果A 或B 為空，而另一個包含單一元素，(A== B).all() 將傳回True。
• 如果 A 和 B 的形狀不同且不可廣播，則會發生錯誤。

替代方法

為了解決這些特殊情況並確保穩健性，請考慮使用專門的NumPy 函數：

• np.array_equal(A, B)：測試相等性，僅考慮同一形狀內的元素.
• np.array_equiv(A, B)：測試是否相等，允許廣播。
• np.allclose(A, B, ...)：比較具有指定容差的元素浮點錯誤。

透過利用這些技術，您可以可靠地比較 NumPy 陣列的相等性，確保程式碼的準確性和一致性。

以上是如何比較 NumPy 數組的相等性？綜合指南的詳細內容。更多資訊請關注PHP中文網其他相關文章！