numpy.allclose()比较具有浮点数的数组

时间:2019-07-16 18:37:37

标签: python numpy floating-point precision floating-accuracy

我有两个numpy数组:

g1 = np.array([3118740.3553, 3520175.8121])
g2 = np.array([3118740.8553, 3520176.3121])

我想使用numpy.allclose()来测试那些数组在浮点精度公差内是否相同

np.allclose(g1, g2, atol=1e-7)

奇怪的是,即使这两个数组之间的差异很大,它也会返回True。为什么?

1 个答案:

答案 0 :(得分:1)

call signature of np.allclose

In [4]: np.allclose?
Signature: np.allclose(a, b, rtol=1e-05, atol=1e-08, equal_nan=False)

请注意,rtol(相对公差)的默认值为1e-05。 只要

abs(a[i] - b[i]) <= rtol * abs(b[i]) + atol

对于所有i = 0, ..., len(a),然后np.allclose返回True。

In [11]: rtol, atol = 1e-05, 1e-7

In [12]: [abs(ai - bi) < rtol * abs(bi) + atol for ai, bi in zip(g1, g2)]
Out[12]: [True, True]

由于g2中的值较大,因此即使较小的rtol也会导致相当大的容差:

In [14]: rtol * g2.min()
Out[14]: 31.187408553

如果不想包含相对公差,则必须将其设置为零以覆盖默认值:

In [13]: np.allclose(g1, g2, rtol=0, atol=1e-7)
Out[13]: False