如何在Pandas的df.query()方法中实现日期列加天数的比较？

要实现arr_date <= dep_date + 7天的筛选，你可以在df.query()的表达式中直接处理日期偏移，以下两种方式都可行：

方法1：直接在query中使用pd.Timedelta #

先确保日期列是datetime类型（若不是，先用pd.to_datetime()转换），再执行筛选：

import pandas as pd

# 构造示例DataFrame
df = pd.DataFrame({
    'Id': [1, 2],
    'arr_date': pd.to_datetime(['2022-01-12', '2022-01-20']),
    'dep_date': pd.to_datetime(['2022-02-01', '2022-03-01'])
})

# 执行筛选
filtered_df = df.query('arr_date <= dep_date + pd.Timedelta(days=7)')

方法2：通过@引用外部时间间隔变量 #

如果需要重复使用该时间偏移，先定义变量再引用会更清晰：

delta = pd.Timedelta(days=7)
filtered_df = df.query('arr_date <= dep_date + @delta')

你的示例数据中两行都满足arr_date早于dep_date+7天的条件，因此筛选结果会返回全部行。

内容的提问来源于stack exchange，提问作者learner