
本教程深入探讨如何在pandas中高效合并两个dataframe,其中一个dataframe的某一时间列需落在另一个dataframe的两个时间列所定义的区间内。我们将摒弃低效的迭代方法,转而采用numpy广播机制,通过矢量化操作显著提升合并大型数据集的性能,并讨论相关的内存管理考量。
在数据分析和处理中,我们经常会遇到需要根据非精确条件(例如时间范围)合并两个DataFrame的情况。一个典型的场景是,我们有一个DataFrame df1,其中包含时间区间的起始 (time_1) 和结束 (time_2),以及另一个DataFrame df2,其中包含单个时间点 (time_3)。我们的目标是,将 df2 中 time_3 落在 df1 任意行的 [time_1, time_2] 区间内的记录与 df1 中对应的行进行合并,并且 df1 的行可能需要被复制多次。
传统方法的局限性
面对这类合并需求,许多开发者可能会自然而然地采用迭代方式。例如,通过嵌套循环遍历 df1 的每一行,然后对 df2 的所有行进行条件筛选,最后使用 pd.concat 或 df.append 将结果逐一拼接。
# 伪代码示例,展示传统方法的思路,实际执行会非常慢 # output_df = pd.DataFrame() # for i in df1.index: # # 筛选 df2 中 time_3 落在 df1.loc[i] 区间内的行 # matching_df2_rows = df2[(df2['time_3'] >= df1.loc[i]['time_1']) & # (df2['time_3'] < df1.loc[i]['time_2'])] # for _, row_df2 in matching_df2_rows.iterrows(): # output_df = output_df.append(pd.concat([df1.loc[i], row_df2]), ignore_index=True)
这种方法虽然直观,但效率极低。Pandas的 loc 索引、循环迭代以及 append 操作(尤其是在循环内部)都会导致大量的性能开销,包括重复的数据复制、内存重新分配和索引重建。对于包含数千甚至数万行以上的数据集,这种方法很快就会变得不可接受。
本文档主要讲述的是基于VC与Matlab的混合编程实现图像的三维显示;介绍了VC++与Matlab混合编程的一般实现方法,并实现对二维影像图的三维效果显示。 MATLAB既是一种直观、高效的计算机语言,同时又是一个科学计算平台。它为数据分析和数据可视化、算法和应用程序开发提供了最核心的数学和高级图形工具。希望本文档会给有需要的朋友带来帮助;感兴趣的朋友可以过来看看
9 基于NumPy广播的高效合并策略
为了克服传统方法的性能瓶颈,我们可以利用NumPy的广播(Broadcasting)机制进行矢量化操作。NumPy广播允许对形状不同的数组执行算术运算,其核心思想是自动扩展较小数组的维度,使其与较大数组的维度兼容,从而实现元素级的操作,而无需显式的python循环。
实现步骤
- 准备数据: 确保参与合并的DataFrame的索引是唯一的。如果DataFrame原有索引不重要,建议在操作前使用 .reset_index(drop=True) 方法重置索引,以避免潜在的索引对齐问题,确保后续 iloc 索引的准确性。
- 转换为NumPy数组并重塑: 将用于比较的时间列转换为NumPy数组。特别是,df1 中的 time_1 和 time_2 列需要被重塑为二维数组(例如 (N, 1) 的形状),而 df2 中的 time_3 列则保持为一维数组((M,) 的形状)。这种形状差异是实现广播的关键。
- 执行广播比较: 利用NumPy的比较运算符(, >=)对重塑后的NumPy数组执行矢量化比较。这将生成一个布尔矩阵,其维度为 (N, M),其中 N 是 df1 的行数,M 是 df2 的行数。矩阵中的每个元素 [i, j] 表示 df1 的第 i 行的时间区间是否包含 df2 的第 j 行的时间点。
- 提取匹配索引: 使用布尔矩阵的 .nonzero() 方法。此方法会返回两个一维NumPy数组 x 和 y,分别包含所有 True 值元素的行索引和列索引。x 中的每个值对应 df1 中匹配的行位置,y 中的每个值对应 df2 中匹配的行位置。
- 合并结果: 最后,利用 iloc 根据 nonzero() 返回的 x 和 y 索引,从原始(或重置索引后)的 df1 和 df2 中选择相应的行。由于 x 和 y 数组的长度相同,我们可以直接将 df1.iloc[x] 和 df2.iloc[y] 通过 pd.concat 沿列方向合并,生成最终的合并结果DataFrame。
示例代码
import pandas as pd import numpy as np # 模拟数据 # DataFrame 1: 包含时间区间 time_1 和 time_2 data1 = { 'time_1': pd.to_datetime(['2023-10-01 04:02:00', '2023-10-01 04:03:00', '2023-10-01 09:00:00']), 'time_2': pd.to_datetime(['2023-10-01 08:29:00', '2023-10-01 08:49:00', '2023-10-01 10:00:00']), 'dummy_data_1': [-245.67, -1772.95, 100.0] } df1 = pd.DataFrame(data1) # DataFrame 2: 包含时间点 time_3 data2 = { 'time_3': pd.to_datetime([ '2023-10-01 06:21:13.238024', '2023-10-01 06:47:19.796628', '2023-10-01 07:37:06.438740', '2023-10-01 08:16:16.995256', '2023-10-01 08:33:53.081095', '2023-10-01 09:30:00.000000' # 此时间点落在df1的第三个区间 ]), 'dummy_data_2': [-131.37, -236.28, 5.92, -134.03, -103.73, 50.0] } df2 = pd.DataFrame(data2) print("DataFrame 1:") print(df1) print("nDataFrame 2:") print(df2) # --- 高效合并逻辑 --- # 1. 确保索引是唯一的。如果不需要保留原有索引,建议重置。 # 这将确保iloc[x]和iloc[y]能正确地按位置索引。 df1_processed = df1.reset_index(drop=True) df2_processed = df2.reset_index(drop=True) # 2. 准备NumPy数组进行广播 # 将time_1和time_2重塑为(N, 1)的二维数组,t3保持(M,)的一维数组 t1 = df1_processed["time_1"].to_numpy()[:, None] t2 = df1_processed["time_2"].to_numpy()[:, None] t3 = df2_processed["time_3"].to_numpy() # 3. 执行广播比较,生成布尔矩阵 # 这里使用开区间 (time_1 < time_3 < time_2)。 # 如果需要左闭右开区间 [time_1, time_2),条件应为 (t1 <= t3) & (t3 < t2)。 # 如果需要闭区间 [time_1, time_2],条件应为 (t1 <= t3) & (t3 <= t2)。 match_matrix = (t1 < t3) & (t3 < t2) # 4. 获取匹配的行索引 # x 对应 df1_processed 的行索引,y 对应 df2_processed 的行索引 x, y = match_matrix.nonzero() # 5. 合并结果 # 使用iloc根据匹配索引从原始DataFrame中选择行,然后沿列方向合并 result_df = pd.concat( [ df1_processed.iloc[x].reset_index(drop=True), # 重置索引以避免重复索引 df2_processed.iloc[y].reset_index(drop=True), # 重置索引以避免重复索引 ], axis=1, ) print("n合并结果:") print(result_df)
输出结果示例:
DataFrame 1: time_1 time_2 dummy_data_1 0 2023-10-01 04:02:00 2023-10-01 08:29:00 -245.67 1 2023-10-01 04:03:00 2023-10-01 08:49:00 -1772.95 2 2023-10-01 09:00:00 2023-10-01 10:00:00 100.00 DataFrame 2: time_3 dummy_data_2 0 2023-10-01