检查日期列是否在 Python 中最近 3 个月的范围内

Question

给定一个小数据集如下：

   id        date
0   1  2020-01-01
1   2  2020-12-02
2   3  2020-09-26
3   4  2020-05-04
4   5  2020-01-05

我想检查 date 是否在 3 个月后 的范围内（从今天开始如果 2020-12-25，则范围将是 [2020-09-25, 2020-12-25])，如果不是，则 return 新列 check 和 N。

预期的结果是：

   id        date check
0   1  2020-01-01     N
1   2  2020-12-02   NaN
2   3  2020-09-26   NaN
3   4  2020-05-04     N
4   5  2020-01-05     N

我怎么能在 Python 中做到这一点？谢谢。

Answer 1

以下解决方案有效：

import datetime 
import dateutil.relativedelta

start_date = (datetime.datetime.now() + dateutil.relativedelta.relativedelta(months=-3)).strftime('%Y-%m-%d')
end_date = datetime.datetime.now().strftime('%Y-%m-%d')

mask = df['date'].between(start_date, end_date, inclusive=True)
df.loc[~mask, 'check'] = 'N'

输出：

   id        date check
0   1  2020-01-01     N
1   2  2020-12-02   NaN
2   3  2020-09-26   NaN
3   4  2020-05-04     N
4   5  2020-01-05     N

检查日期列是否在 Python 中最近 3 个月的范围内

Check if date column is in the range of last 3 months in Python

datetime

date

python-3.x

python-dateutil