Python numpy testing.assert_array_almost_equal用法及代碼示例

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

用法:

testing.assert_array_almost_equal(x, y, decimal=6, err_msg='', verbose=True)

如果兩個對象不等於所需的精度，則引發 AssertionError。

注意

建議使用 assert_allclose 、 assert_array_almost_equal_nulp 或 assert_array_max_ulp 之一代替此函數，以獲得更一致的浮點比較。

該測試驗證相同的形狀以及actual 和desired 的元素是否滿足。

abs(desired-actual) < 1.5 * 10**(-decimal)

這是一個比最初記錄的更寬鬆的測試，但與實際實現所做的四舍五入的變化一致。在形狀不匹配或值衝突時引發異常。與 numpy 中的標準用法相比，NaNs 像數字一樣進行比較，如果兩個對象的 NaNs 在相同的位置，則不會引發斷言。

參數：

x： array_like

要檢查的實際對象。

y： array_like

期望的、預期的對象。

decimal： 整數，可選

所需精度，默認為 6。

err_msg： str，可選

失敗時要打印的錯誤消息。

verbose： 布爾型，可選

如果為 True，則將衝突值附加到錯誤消息中。

拋出：

AssertionError

如果實際和期望不等於指定的精度。

例子：

第一個斷言不會引發異常

>>> np.testing.assert_array_almost_equal([1.0,2.333,np.nan],
...                                      [1.0,2.333,np.nan])

>>> np.testing.assert_array_almost_equal([1.0,2.33333,np.nan],
...                                      [1.0,2.33339,np.nan], decimal=5)
Traceback (most recent call last):
    ...
AssertionError:
Arrays are not almost equal to 5 decimals

Mismatched elements: 1 / 3 (33.3%)
Max absolute difference: 6.e-05
Max relative difference: 2.57136612e-05
 x: array([1.     , 2.33333,     nan])
 y: array([1.     , 2.33339,     nan])

>>> np.testing.assert_array_almost_equal([1.0,2.33333,np.nan],
...                                      [1.0,2.33333, 5], decimal=5)
Traceback (most recent call last):
    ...
AssertionError:
Arrays are not almost equal to 5 decimals

x and y nan location mismatch:
 x: array([1.     , 2.33333,     nan])
 y: array([1.     , 2.33333, 5.     ])