numpy.isclose#

numpy.isclose(a, b, rtol=1e-05, atol=1e-08, equal_nan=False)[source]#

返回一个布尔数组,表示两个数组在给定容差范围内是否元素级相等。

容差值是正数,通常是非常小的数字。相对差值 (rtol * abs(b)) 和绝对差值 atol 被加在一起,用于与 ab 之间的绝对差值进行比较。

警告

默认的 atol 不适用于比较数量级远小于1的数字(参见注释)。

参数:
a, barray_like

要比较的输入数组。

rtolarray_like

相对容差参数(参见注释)。

atolarray_like

绝对容差参数(参见注释)。

equal_nanbool

是否将 NaN 视为相等。如果为 True,输出数组中 a 中的 NaN 将被视为与 b 中的 NaN 相等。

返回:
yarray_like

返回一个布尔数组,表示 ab 在给定容差范围内是否相等。如果 ab 都是标量,则返回单个布尔值。

另请参见

allclose
math.isclose

注释

对于有限值,isclose 使用以下等式来测试两个浮点值是否等效。

absolute(a - b) <= (atol + rtol * absolute(b))

与内置的 math.isclose 不同,上述等式在 ab 之间不对称——它假设 b 是参考值——因此 isclose(a, b) 可能与 isclose(b, a) 不同。

当参考值 b 的数量级小于一时,atol 的默认值不适用。例如,`a = 1e-9` 和 `b = 2e-9` 不太可能被认为是“接近”的,但在默认设置下 `isclose(1e-9, 2e-9)` 却是 `True`。请务必根据具体用例选择 atol,特别是为了定义一个阈值,在该阈值之下,a 中的非零值将被视为“接近”b 中的一个非常小或零值。

isclose 不适用于非数值数据类型。bool 在此目的下被视为数值数据类型。

示例

>>> import numpy as np
>>> np.isclose([1e10,1e-7], [1.00001e10,1e-8])
array([ True, False])
>>> np.isclose([1e10,1e-8], [1.00001e10,1e-9])
array([ True, True])
>>> np.isclose([1e10,1e-8], [1.0001e10,1e-9])
array([False,  True])
>>> np.isclose([1.0, np.nan], [1.0, np.nan])
array([ True, False])
>>> np.isclose([1.0, np.nan], [1.0, np.nan], equal_nan=True)
array([ True, True])
>>> np.isclose([1e-8, 1e-7], [0.0, 0.0])
array([ True, False])
>>> np.isclose([1e-100, 1e-7], [0.0, 0.0], atol=0.0)
array([False, False])
>>> np.isclose([1e-10, 1e-10], [1e-20, 0.0])
array([ True,  True])
>>> np.isclose([1e-10, 1e-10], [1e-20, 0.999999e-10], atol=0.0)
array([False,  True])