尝试加入2个数据框,并将某些数据作为数组存储在一个单元格中
问题内容:
我有以下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]
问题答案:
联合会
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]