如何筛选DataFrame中符合条件的行:ATime不早于BTime超1分钟或更晚
问题描述
我有一个包含Atime和BTime两个datetime类型列的DataFrame,需要筛选出满足以下条件的行:
- Atime早于BTime,但两者的时间差不超过1分钟
- Atime晚于BTime
原始数据
| Atime | BTime |
|---|---|
| 06/01/2017 19:58:01 | 06/01/2017 20:00:00 |
| 06/01/2017 19:59:01 | 06/01/2017 20:00:00 |
| 06/01/2017 20:00:01 | 06/01/2017 20:00:00 |
期望结果
| Atime | BTime |
|---|---|
| 06/01/2017 19:59:01 | 06/01/2017 20:00:00 |
| 06/01/2017 20:00:01 | 06/01/2017 20:00:00 |
解决方案
可以用Pandas的时间差计算功能实现筛选,步骤如下:
- 转换列为datetime类型:如果
Atime和BTime尚未是datetime格式,先进行转换:
import pandas as pd # 假设原始数据存储在df中 df['Atime'] = pd.to_datetime(df['Atime'], format='%m/%d/%Y %H:%M:%S') df['BTime'] = pd.to_datetime(df['BTime'], format='%m/%d/%Y %H:%M:%S')
- 计算时间差并筛选行:通过计算
BTime与Atime的时间差,设置条件筛选符合要求的行:
# 计算时间差,转换为分钟单位 time_diff = (df['BTime'] - df['Atime']).dt.total_seconds() / 60 # 筛选条件:时间差≤1分钟(包含Atime晚于BTime的情况,此时time_diff为负数) filtered_df = df[time_diff <= 1]
执行上述代码后,filtered_df即为符合期望的结果。
内容的提问来源于stack exchange,提问作者Andy
相关产品推荐
相关产品推荐

