NumPy浮点数数组的精确比较：使用isclose处理精度差异

本教程详细讲解了在NumPy中进行浮点数数组精确比较的方法，着重介绍`numpy.isclose`函数。针对浮点数计算固有的精度问题，`isclose`允许通过设置绝对容差（`atol`）和相对容差（`rtol`）来判断两个浮点数是否在可接受的误差范围内相等，从而有效解决直接等值比较的局限性。

在科学计算和数据分析中，NumPy是处理数值数组的核心库。然而，当涉及到浮点数（float）的比较时，开发者经常会遇到一个常见挑战：由于计算机内部表示浮点数的机制，某些看起来应该相等的值，在直接使用==运算符进行比较时，结果却为False。这并非NumPy特有的问题，而是所有基于二进制表示的浮点数运算的固有特性。例如，0.1 + 0.2并不严格等于0.3。

考虑以下场景，我们有一个NumPy数组，其中包含浮点数，我们希望将其与另一个数组或特定值进行比较，判断它们是否“足够接近”，而非“精确相等”：

import numpy as np

e = np.array([0.8292222222222225, 0.1310000000000003])
print(f"原始数组元素 e[0]: {e[0]}")
print(f"直接比较 e[0] == 0.829225: {e[0] == 0.829225}")

输出结果会是：

原始数组元素 e[0]: 0.8292222222222225
直接比较 e[0] == 0.829225: False

尽管0.8292222222222225和0.829225在肉眼看来非常接近，但由于它们在二进制表示上的微小差异，直接的等值比较返回了False。在这种情况下，我们不能通过简单地修改打印选项（如np.set_printoptions）来解决问题，因为这仅影响显示，不改变底层数值。我们真正需要的是一种能够考虑数值“接近程度”的比较机制。

解决方案：使用numpy.isclose函数

NumPy提供了一个专门用于浮点数“近似相等”比较的函数：numpy.isclose。此函数允许用户定义一个可接受的误差范围（容差），如果两个数值在此容差范围内，则认为它们是相等的。这正是解决浮点数比较难题的理想工具。

numpy.isclose的函数签名如下： numpy.isclose(a, b, rtol=1e-05, atol=1e-08, equal_nan=False)

其中，最关键的参数是：

火山写作

字节跳动推出的中英文AI写作、语法纠错、智能润色工具，是一款集成创作、润色、纠错、改写、翻译等能力的中英文 AI 写作助手。

167

查看详情

• a, b: 待比较的两个数组（或标量）。
• rtol (relative tolerance): 相对容差。它表示与较大值相比允许的最大相对差异。默认值为1e-05。
• atol (absolute tolerance): 绝对容差。它表示允许的最大绝对差异。默认值为1e-08。
• equal_nan: 如果设置为True，则两个NaN（Not a Number）值将被视为相等。默认值为False。

isclose函数会根据以下条件判断两个元素a和b是否接近： abs(a - b) <= (atol + rtol * abs(b))

这意味着，当两个值之间的绝对差值小于或等于绝对容差加上相对容差乘以b的绝对值时，它们被认为是接近的。

理解容差参数：rtol和atol

选择合适的容差值是有效使用isclose的关键。

1. 绝对容差 (atol): atol定义了一个固定的小数值，表示两个数之间允许的最大绝对差值。它适用于比较接近零的数值，或者当误差的绝对大小是恒定且已知时。例如，如果你知道你的测量精度是+/- 0.001，那么atol=1e-3可能是一个合适的选择。

2. 相对容差 (rtol): rtol定义了一个百分比或比例，表示两个数之间允许的最大相对差异。它更适用于比较大小差异很大的数值。例如，比较1和1.00001，以及10000和10000.00001时，相对容差能更好地捕捉“接近”的含义。当数值远离零时，相对容差通常比绝对容差更有用。

在实际应用中，通常会同时使用atol和rtol，以覆盖不同数量级的数值比较场景。

示例代码

让我们通过具体的例子来演示numpy.isclose如何解决浮点数比较问题。我们仍然使用之前定义的数组e，并引入另一个数组b进行比较。

import numpy as np

a = np.array([0.8292222222222225, 0.1310000000000003])
b = np.array([0.8293, 0.132])

print(f"数组 a: {a}")
print(f"数组 b: {b}\n")

# 使用不同的绝对容差 (atol) 进行比较
print("使用 atol=1e-3 进行比较:")
print(np.isclose(a, b, atol=1e-3)) 

print("\n使用 atol=1e-4 进行比较:")
print(np.isclose(a, b, atol=1e-4)) 

print("\n使用 atol=1e-5 进行比较:")
print(np.isclose(a, b, atol=1e-5))

# 如果我们希望比较 e[0] 和 0.829225
e_single = np.array([0.8292222222222225])
target_single = np.array([0.829225])
print(f"\n比较 {e_single[0]} 和 {target_single[0]}:")
print(f"使用 atol=1e-5: {np.isclose(e_single, target_single, atol=1e-5)}")
print(f"使用 atol=1e-6: {np.isclose(e_single, target_single, atol=1e-6)}")

输出结果：

数组 a: [0.82922222 0.131   ]
数组 b: [0.8293 0.132 ]

使用 atol=1e-3 进行比较:
[ True  True]

使用 atol=1e-4 进行比较:
[ True False]

使用 atol=1e-5 进行比较:
[False False]

比较 0.8292222222222225 和 0.829225:
使用 atol=1e-5: [ True]
使用 atol=1e-6: [False]

从上述示例可以看出，通过调整atol参数，我们可以精确控制比较的严格程度。当atol=1e-3时，a和b的两个对应元素都被认为是接近的。但随着atol减小到1e-4，第二个元素（0.131和0.132）的差异0.001超出了容差范围，因此被判定为不接近。这直接解决了用户提出的“比较0.8292222222222225与0.829225为True”的需求，只需设置合适的atol即可。

注意事项与最佳实践

1. isclose是首选方法: 在NumPy中进行浮点数近似比较时，numpy.isclose是标准且推荐的方法。它比手动四舍五入或截断数值再比较更加健壮和灵活。
2. 选择合适的容差值: rtol和atol的选择取决于具体的应用场景和所需的精度。没有一个通用的“最佳”容差值。通常需要根据数据的特性、预期的误差范围以及业务需求进行试验和调整。
3. 考虑numpy.allclose:

以上就是NumPy浮点数数组的精确比较：使用isclose处理精度差异的详细内容，更多请关注php中文网其它相关文章！