小编典典

尝试加入2个数据框,并将某些数据作为数组存储在一个单元格中

python

我有以下3个数据框:

第一数据框:

DF1:

iID data1 data2

10  blue  green

11  red   teal

第二个数据框:

DF2:

iID rH repH

10  50 60

10  60 70

11  70 50

(DF2每个iID可以有1或2行)

我希望我的输出DF在一个单元格中有一个数组用于rH和repH

输出将是这样的:

输出DF:

iID data1 data2 rH      repH

10  blue  green [50,60] [60,70]

11  red   teal  [70]    [50]

阅读 150

收藏
2021-01-20

共1个答案

小编典典

联合会

df1.merge(df2.groupby('iID').agg(lambda x : x.tolist()).reset_index())
Out[144]: 
   iID data1  data2        rH      repH
0   10  blue  green  [50, 60]  [60, 70]
1   11   red   teal      [70]      [50]
2021-01-20