根据不同条件加入 pandas 数据框

Question

我在 Python 中使用 Pandas，我想加入 2 个数据框。我的第一个数据框是：

id	var	date
1	ABCD	2019-01-01
1	ABCD	2017-06-01
1	ABCD	2016-06-01
2	ABCD	2016-01-01

我要加入的数据框是：

var	date
ABCD	2019-01-01
ABCD	2017-06-01
FJKL	2016-06-01
ABCD	2016-01-01

我想通过var 和date 加入第一个数据帧，但条件是第二个数据帧的日期必须小于第一个数据帧的日期。如果我在 sql 中，我会这样做：

   select 
   a.*,
   b.* 
   from fisrt_data a 
   join second_data b on a.var = b.var and a.date > b.date

有什么想法可以用 pandas 来实现吗？

Answer 1

有merge和query：

(df1.merge(df2, on=['var'], suffixes=['_a','_b'])
    .query('date_a > date_b')
)

输出：

   id    var       date_a      date_b
1    1  ABCD   2019-01-01  2017-06-01
2    1  ABCD   2019-01-01  2016-01-01
5    1  ABCD   2017-06-01  2016-01-01
8    1  ABCD   2016-06-01  2016-01-01