根据另一个数据帧的日期范围进行过滤

Question

我有两个 pandas 数据框，如下所示：

df1:

id  date        item
3   2015-11-23  B
3   2015-11-23  A
3   2016-05-11  C
3   2017-02-01  C
3   2018-07-12  E
4   2014-05-11  C
4   2015-02-01  C
4   2018-07-12  E

df2

id  start       end            
3   2016-05-11  2017-08-30
4   2015-01-11  2017-08-22

我想剪切 df1 ，这样我只保留 df1 中属于df2 中给出的日期范围：

id  date        item
3   2016-05-11  C
3   2017-02-01  C
4   2015-02-01  C

实际上，df1 和 df2 有数百万行，因此，我无法使用 for 循环等进行任何快速修复。我大致了解如何使用 groupby by id，但恐怕到目前为止我所有的尝试都失败了。

先感谢您！

Answer 1

基本方法是构建一个包含该 id 的所有可能事件的数据帧。然后，您可以过滤该事件是否在两个日期之间。

df3 = df1.merge(df2, how='inner', left_on='id', right_on='id')

df3[(df3['date'] <= df3['end']) & (df3['date'] >= df3['date'])]