python Pandas中的条件日期加入

Question

我有两个熊猫数据框matches，带有列（match_id，team_id，date，...）和teams_att，带有列（id，team_id，date，@ 987654329@, ...)。 我想加入最接近matches.date 的matches.team_id = teams_att.team_id 和teams_att.date 上的两个数据框

例子

匹配

match_id    team_id     date
1           101         2012-05-17
2           101         2014-07-11
3           102         2010-05-21
4           102         2017-10-24

teams_att

id  team_id     date        overall_rating
1   101         2010-02-22  67
2   101         2011-02-22  69
3   101         2012-02-20  73
4   101         2013-09-17  79
5   101         2014-09-10  74
6   101         2015-08-30  82
7   102         2015-03-21  42
8   102         2016-03-22  44

期望的结果

match_id    team_id     matches.date    teams_att.date      overall_rating
1           101         2012-05-17      2012-02-20          73
2           101         2014-07-11      2014-09-10          74
3           102         2010-05-21      2015-03-21          42
4           102         2017-10-24      2016-03-22          44

Answer 1

我们使用merge_asof (Please check Scott's answer, that is the right way for solving this type problem :-) cheers)

g1=df1.groupby('team_id')
g=df.groupby('team_id')

l=[]
for x in [101,102]:
   l.append(pd.merge_asof(g.get_group(x),g1.get_group(x),on='date',direction ='nearest'))


pd.concat(l)
Out[405]: 
   match_id  team_id_x       date  id  team_id_y  overall_rating
0         1        101 2012-05-17   3        101              73
1         2        101 2014-07-11   5        101              74
0         3        102 2010-05-21   7        102              42
1         4        102 2017-10-24   8        102              44

Answer 2

您可以将merge_asof 与by 和direction 参数一起使用：

pd.merge_asof(matches.sort_values('date'), 
              teams_att.sort_values('date'), 
              on='date', by='team_id', 
              direction='nearest')

输出：

   match_id  team_id       date  id  overall_rating
0         3      102 2010-05-21   7              42
1         1      101 2012-05-17   3              73
2         2      101 2014-07-11   5              74
3         4      102 2017-10-24   8              44