如何使用python panda有效地基于日期重叠范围对行进行子集设置

Question

我的数据帧有两个日期类型列：开始和结束(yyyy-mm-dd(。

这是我的数据框架：

import pandas as pd
import datetime 
data=[["2016-10-17","2017-03-08"],["2014-08-17","2016-09-08"],["2014-01-01","2015-01-01"],["2017-12-20","2019-01-01"]]
df=pd.DataFrame(data,columns=['start','end'])
df['start'] =  pd.to_datetime(df['start'], format='%Y-%m-%d')
df['end'] =  pd.to_datetime(df['end'], format='%Y-%m-%d')
start        end
0 2016-10-17 2017-03-08
1 2014-08-17 2016-09-08
2 2014-01-01 2015-01-01
3 2017-12-20 2019-01-01

我有参考的开始和结束日期如下。

ref_start=datetime.date(2015, 9, 20)
ref_end=datetime.date(2017,1,31)
print(ref_start,ref_end)
2015-09-20 2017-01-31

如果行的开始和结束日期范围与引用开始和结束时间重叠，我希望对行进行子集设置。由于开始和结束日期范围与参考日期范围不重叠，因此未选择第三行和第四行(2015-09-20 ~ 2017-01-31(

所以我想要的结果是这样的：

start         end
0  2016-10-17  2017-03-08
1  2014-08-17  2016-09-08

为此，我考虑在此基础上使用以下代码：python中有效的日期范围重叠计算？

df[(max(df['start'],ref_start)>min(df['end'],ref_end))]

然而，它不起作用。有什么方法可以有效地获得期望的结果吗？

Answer 1

我在职业生涯早期学到的一个技巧是我所说的"跨越日期"：将一个范围的开始与另一个范围结束进行比较。

# pd.Timestamp can do everything that datetime/date does and some more
ref_start = pd.Timestamp(2015, 9, 20)
ref_end = pd.Timestamp(2017,1,31)
# Compare the start of one range to the end of another and vice-versa
# Made into a separate variable for reability
cond = (ref_start <= df['end']) & (ref_end >= df['start'])
df[cond]