Python Pandas：高效比较结构相似但列名与数据有异的DataFrame

背景与挑战

在数据分析和处理过程中，我们经常会遇到需要比较两个DataFrame的场景。这些DataFrame可能来源于不同的数据源，尽管它们在逻辑上代表相同的信息，但可能存在以下挑战：

1. 列名差异： 相同的列可能在不同的DataFrame中拥有略微不同的名称（例如，Town A vs Town A_U）。
2. 数据值差异： 对应单元格中的数据可能存在差异。
3. NaN 值处理： 两个DataFrame中都可能存在缺失值（NaN），在比较时需要将它们视为相等。
4. 输出要求： 期望得到一个布尔型的DataFrame，直观地显示哪些单元格是匹配的（True）或不匹配的（False）。

传统上，直接使用 df1 == df2 进行比较可能无法正确处理 NaN 值（NaN == NaN 结果为 False），且对列名差异不友好。因此，我们需要一种更健壮的方法来应对这些复杂性。

核心解决方案：使用 numpy.isclose 进行精确比较

numpy.isclose 函数是解决此类DataFrame比较问题的强大工具。它能够对两个数组（或DataFrame）进行元素级别的比较，并提供了灵活的参数来处理浮点数精度和 NaN 值。

1. 理解 numpy.isclose

numpy.isclose(a, b, rtol=1e-05, atol=1e-08, equal_nan=False) 函数用于判断两个数组 a 和 b 中的对应元素是否“接近”。对于我们的精确比较需求，关键参数配置如下：

立即学习“Python免费学习笔记（深入）”；

• a, b：要比较的两个数组或DataFrame。
• equal_nan=True：这是最关键的参数。当设置为 True 时，NaN 值将被视为相等。这意味着如果 a 和 b 中对应位置都是 NaN，则 isclose 会返回 True。
• rtol=0, atol=0：这两个参数分别代表相对容忍度和绝对容忍度，用于浮点数的近似比较。将它们都设置为 0 可以确保进行严格的、精确的比较，即只有当两个数值完全相等时才返回 True（除了 NaN 的特殊处理）。

2. 数据准备（隐含前提与注意事项）

在使用 numpy.isclose 直接比较两个DataFrame之前，需要确保以下前提：

• 形状一致： 两个DataFrame (dfa 和 dfb) 必须具有相同的行数和列数。numpy.isclose 执行的是基于位置的元素比较。
• 列的逻辑对齐： 尽管列名可以不同，但它们在DataFrame中的 顺序 必须与它们所代表的逻辑含义相符。例如，如果 dfa 的第一列是“城镇A”，dfb 的第一列是“城镇A_U”，且它们都指代同一概念，那么这种直接比较是有效的。如果列的逻辑顺序被打乱，则需要进行预处理，例如：
  • 统一列名： 将 dfb 的列名重命名为与 dfa 相同的名称。
  • 重新排序列： 确保 dfb 的列顺序与 dfa 的列顺序一致。

本教程的示例假设列的顺序已经逻辑对齐。

3. 执行比较操作

一旦DataFrame准备就绪，就可以直接将它们作为参数传递给 numpy.isclose：

import pandas as pd
import numpy as np

# 假设dfa和dfb是已经加载的DataFrame
# dfa:
#    Town A  Town B  Town C
# 0     NaN     NaN     1.0
# 1     3.0    11.0     NaN
# 2     NaN     3.0     NaN

# dfb:
#    Town A_U  Town B  Town C g
# 0       NaN     NaN       1.0
# 1       3.0     NaN       NaN
# 2       NaN     4.0       NaN

comparison_array = np.isclose(dfa, dfb, equal_nan=True, rtol=0, atol=0)

comparison_array 将是一个布尔型的NumPy数组，其形状与输入DataFrame相同，每个元素表示对应位置的值是否匹配。

LuckyCola工具库

LuckyCola工具库是您工作学习的智能助手，提供一系列AI驱动的工具，旨在为您的生活带来便利与高效。

133

查看详情

4. 结果整合为 Pandas DataFrame

为了获得与原始DataFrame结构相似的布尔型输出，我们需要将 comparison_array 转换回 Pandas DataFrame，并保留原始的索引和列名（通常选择其中一个输入DataFrame的索引和列名，例如 dfa）。

output_df = pd.DataFrame(comparison_array, index=dfa.index, columns=dfa.columns)

示例代码

以下是一个完整的示例，演示如何比较两个具有不同列名和 NaN 值的DataFrame，并生成一个指示匹配情况的布尔型DataFrame。

import pandas as pd
import numpy as np

# 示例数据：DataFrame A
data_a = {
    'Town A': [np.nan, 3, np.nan],
    'Town B': [np.nan, 11, 3],
    'Town C': [1, np.nan, np.nan]
}
dfa = pd.DataFrame(data_a, index=[1, 2, 3])

print("DataFrame A:")
print(dfa)
print("-" * 30)

# 示例数据：DataFrame B (列名略有不同，部分值不同)
data_b = {
    'Town A_U': [np.nan, 3, np.nan],
    'Town B': [np.nan, np.nan, 4], # 注意这里 Town B 的第2行和第3行与dfa不同
    'Town C g': [1, np.nan, np.nan]
}
dfb = pd.DataFrame(data_b, index=[1, 2, 3])

print("DataFrame B:")
print(dfb)
print("-" * 30)

# 核心比较逻辑
# 使用 numpy.isclose 进行元素级比较，设置 equal_nan=True 确保 NaN 与 NaN 匹配
# rtol=0, atol=0 确保进行精确匹配
comparison_array = np.isclose(dfa, dfb, equal_nan=True, rtol=0, atol=0)

# 将布尔数组转换为 DataFrame，保留 dfa 的索引和列名
output_df = pd.DataFrame(comparison_array, index=dfa.index, columns=dfa.columns)

print("比较结果 DataFrame (True 表示匹配，False 表示不匹配):")
print(output_df)

输出结果：

DataFrame A:
   Town A  Town B  Town C
1     NaN     NaN     1.0
2     3.0    11.0     NaN
3     NaN     3.0     NaN
------------------------------
DataFrame B:
   Town A_U  Town B  Town C g
1       NaN     NaN       1.0
2       3.0     NaN       NaN
3       NaN     4.0       NaN
------------------------------
比较结果 DataFrame (True 表示匹配，False 表示不匹配):
   Town A  Town B  Town C
1    True    True    True
2    True   False    True
3    True   False    True

从输出可以看出，Town B 列的第2行 (11.0 vs NaN) 和第3行 (3.0 vs 4.0) 均被正确识别为不匹配（False）。

重要注意事项

1. 列的对齐与命名：numpy.isclose 是基于位置进行比较的。如果两个DataFrame的列名不同，但它们的 顺序 已经能够确保对应列的语义一致性，那么可以直接使用。否则，在执行 isclose 之前，必须对其中一个DataFrame的列进行重命名或重新排序，使其与另一个DataFrame的列在逻辑上和位置上对齐。例如，可以使用 dfb.columns = dfa.columns 来统一列名（前提是列的顺序已匹配），或者使用 dfb = dfb[dfa.columns] 来重新排序列。

2. NaN 值的处理： 始终记住在需要将 NaN 视为相等时，设置 equal_nan=True。这是 numpy.isclose 相较于 df1 == df2 的一个显著优势。

3. 精确度控制：rtol=0 和 atol=0 确保了比较的绝对精确性。如果您的场景允许浮点数存在微小差异（例如，由于计算误差），您可以根据需求调整 rtol 和 atol 的值。

4. 性能考量： 由于 numpy.isclose 是基于NumPy数组操作的，它在处理大型DataFrame时通常非常高效，性能优于纯Python循环或其他逐元素比较方法。

总结

通过巧妙地利用 numpy.isclose 函数，我们可以高效且准确地比较两个结构相似但存在列名差异和 NaN 值的Pandas DataFrame。关键在于理解 equal_nan=True 参数的作用，并确保在比较前DataFrame的列已进行逻辑对齐。这种方法不仅提供了清晰的布尔型比较结果，而且在处理大规模数据时保持了良好的性能。掌握这一技巧，将极大地提升您在数据质量检查和数据差异分析方面的能力。

以上就是Python Pandas：高效比较结构相似但列名与数据有异的DataFrame的详细内容，更多请关注php中文网其它相关文章！