使用Pandas交叉合并高效检查DataFrame值范围

在数据分析和处理中，我们经常会遇到需要比较两个DataFrame中数据的情况，例如检查一个DataFrame中的数值是否落在另一个DataFrame定义的某个区间内。对于大规模数据集，传统的行迭代方法效率低下，难以满足性能要求。本文将详细阐述如何利用Pandas的cross merge功能，以向量化的方式高效完成此类任务。

1. 场景描述与数据准备

假设我们有两个Pandas DataFrame，df1 包含一系列时间点及其对应的 High、Mid、Low 值，而 df2 包含一些时间段 (Start) 以及对应的 Top 和 Bottom 边界值。我们的目标是，对于 df2 中的每一行，统计 df1 中有多少行满足以下条件：

1. df1 的 Date 不等于 df2 的 Start。
2. df1 的 High 值大于 df2 的 Bottom 值。
3. df1 的 Mid 和 Low 中的最大值小于 df2 的 Top 值。

最终，将匹配的数量存储在 df2 的一个新列 Match 中。

以下是示例数据结构：

import pandas as pd
import numpy as np

# 示例 df1
data1 = {
    'Date': pd.to_datetime(['2023-08-03 00:00:00', '2023-08-03 12:00:00', '2023-08-04 12:00:00',
                            '2023-08-05 00:00:00', '2023-08-05 20:00:00', '2023-11-26 20:00:00',
                            '2023-11-28 00:00:00', '2023-11-28 16:00:00', '2023-11-29 08:00:00',
                            '2023-11-30 04:00:00']),
    'High': [29249.8, 29395.8, 29305.2, 29099.9, 29061.6, 37732.1, 37341.4, 38390.7, 38419.0, 38148.9],
    'Mid': [29136.6, 29228.1, 29250.1, 29045.3, 29047.1, 37469.9, 37138.2, 38137.2, 38136.3, 37800.1],
    'Low': [29152.3, 29105.0, 29137.1, 29073.0, 29044.0, 37370.0, 37254.1, 37534.4, 38112.0, 38040.0]
}
df1 = pd.DataFrame(data1)
df1.index = [1, 4, 10, 13, 18, 696, 703, 707, 711, 716]

# 示例 df2
data2 = {
    'Start': pd.to_datetime(['2023-11-28 00:00:00', '2023-11-24 12:00:00']),
    'Top': [37341.4, 38432.9],
    'Bottom': [37138.2, 37894.4]
}
df2 = pd.DataFrame(data2)

print("df1:")
print(df1)
print("\ndf2:")
print(df2)

2. 迭代式解决方案及其局限性

一个直观但效率低下的方法是使用循环迭代 df1 的每一行，并在 df2 上应用条件筛选：

百度智能云·曦灵

百度旗下的AI数字人平台

83

查看详情

# 初始化 df2 的 'Match' 列
df2['Match'] = 0

# 迭代式解决方案 (低效)
# for idx in df1.index:
#     df2.loc[
#         (df2.Start !=  df1.at[idx, 'Date']) &
#         (df2.Bottom < df1.at[idx, 'High']) &
#         (df2.Top > df1.loc[idx, ['Mid', 'Low']].max()),
#         'Match'] += 1
# print("\n迭代式结果 (注释掉以避免运行):")
# print(df2)

这种方法通过Python循环逐行处理，每迭代一次都会对 df2 进行一次条件判断和赋值操作。当 df1 或 df2 的行数较大时，这种逐行操作的开销会非常显著，导致性能瓶颈。Pandas的设计理念是鼓励向量化操作，避免显式的Python循环。

3. 基于交叉合并的高效解决方案

为了避免迭代，我们可以利用Pandas的merge函数进行cross merge（交叉合并），它会创建两个DataFrame所有行组合的笛卡尔积。然后，我们可以对这个合并后的巨大DataFrame进行一次性、向量化的条件筛选。

# 确保df2的原始索引在合并后可用，以便后续计数
df2_temp = df2.reset_index()

# 执行交叉合并，创建所有可能的行组合
# 合并后的DataFrame将包含df2的每一行与df1的每一行的组合
merged_df = df2_temp.merge(df1, how='cross')

# 应用所有筛选条件
# 注意：x['Mid', 'Low'].max(axis=1) 用于获取每行Mid和Low的最大值
filtered_matches = merged_df.loc[
    (merged_df.Start != merged_df.Date) &
    (merged_df.Bottom < merged_df.High) &
    (merged_df.Top > merged_df[['Mid', 'Low']].max(axis=1))
]

# 统计每个原始df2索引的匹配数量
# value_counts('index') 会统计在filtered_matches中，每个原始df2索引出现的次数
match_counts = filtered_matches.value_counts('index')

# 将匹配数量重新索引回df2的原始索引，未匹配的填0
# reindex确保所有df2的原始索引都存在，即使它们没有匹配项
df2['Match'] = match_counts.reindex(df2.index, fill_value=0)

print("\n交叉合并解决方案结果:")
print(df2)

代码解析：

1. df2_temp = df2.reset_index(): 在进行交叉合并之前，我们先将 df2 的索引重置，将其转换为一个普通列（名为index）。这样做是为了在合并后能够识别出每个匹配项是属于 df2 的哪一行，因为value_counts需要一个列名来计数。
2. merged_df = df2_temp.merge(df1, how='cross'): 这是核心步骤。它将 df2_temp 和 df1 进行交叉合并，生成一个包含 len(df2) * len(df1) 行的新DataFrame。这个DataFrame的每一行都是 df2 的一行与 df1 的一行的组合。
3. filtered_matches = merged_df.loc[...]: 对合并后的 merged_df 应用所有条件进行筛选。由于 merged_df 包含了所有列，我们可以直接使用列名进行向量化比较。
   • merged_df.Start != merged_df.Date: 比较日期不相等。
   • merged_df.Bottom < merged_df.High: 比较 Bottom 和 High。
   • merged_df.Top > merged_df[['Mid', 'Low']].max(axis=1): 这一步很关键，它首先从 Mid 和 Low 列中找出每行的最大值（axis=1 表示按行操作），然后与 Top 进行比较。
4. match_counts = filtered_matches.value_counts('index'): 在筛选后的 filtered_matches 中，我们统计原始 df2 索引（即 df2_temp 中 index 列的值）出现的次数。这直接给出了每个 df2 行的匹配数量。
5. df2['Match'] = match_counts.reindex(df2.index, fill_value=0): 最后，我们将统计结果赋值给 df2 的 Match 列。reindex 函数确保 df2 的所有原始索引都得到处理。如果某个 df2 的行没有匹配项（即其索引不在 match_counts 中），reindex 会用 fill_value=0 填充其 Match 值。

4. 性能考量与注意事项

• 内存消耗: cross merge 的主要缺点是内存消耗。合并后的DataFrame的大小是 len(df1) * len(df2)。如果 df1 和 df2 都非常大，例如各有数十万行，那么合并后的DataFrame将会有数万亿行，这会迅速耗尽系统内存。因此，在使用 cross merge 之前，务必评估数据集的大小。对于本例中的小数据集，这种方法非常高效。
• 适用场景: cross merge 最适合于其中一个或两个DataFrame相对较小，或者总的笛卡尔积在可接受内存范围内的场景。
• 条件复杂性: 这种方法能够优雅地处理复杂的组合条件，所有条件都可以一次性地应用到合并后的DataFrame上，充分利用Pandas的向量化能力。
• 替代方案: 对于超大型数据集，如果 cross merge 不可行，可能需要考虑其他策略，例如：
  • 使用Dask等分布式计算库。
  • 将问题分解为更小的批次处理（分块处理）。
  • 利用数据库的Join操作进行预处理。
  • 如果条件允许，考虑使用Cython或Numba等工具优化Python循环。

总结

通过使用Pandas的cross merge功能，我们可以将两个DataFrame之间的复杂条件比较和匹配计数任务，从低效的迭代模式转换为高性能的向量化操作。这种方法显著提升了代码的执行效率和可读性，是处理中小型数据集交叉比较问题的强大工具。然而，在使用时必须密切关注其内存消耗，以避免潜在的性能问题。正确评估数据集规模并选择合适的工具，是高效数据处理的关键。

以上就是使用Pandas交叉合并高效检查DataFrame值范围的详细内容，更多请关注php中文网其它相关文章！