从另一个数据框架中撤出一个数据时，我被编写了Python代码

Question

我有两个数据帧。

df1 = pd.DataFrame({'vin':['aaa','aaa','aaa','bbb','ccc','ccc','ddd','eee','eee','fff'],'module':['ABS','ABS','IPMA','BCCM','HPOC','ABS','ABS','HPOC','ABS','ABS']})


df2 = pd.DataFrame({'vin':['aaa','bbb','ccc','ddd','eee','fff']})

因此，基本上，在DF2中，我想从DF1用相应的列“ VIN”从DF1中拉出“模块”列的值，但挑战是我希望一个由一个逗号分隔的单元格中的所有值。 我尝试了以下命令。

df_merge = pd.merge(df2, df1[['module','vin']], on ='vin', how ='left')

现在，这条代码行的问题是，它正在以我不想要的多行中拉数据。

我的预期输出将是这样： -

df2 = pd.DataFrame({'vin':['aaa','bbb','ccc','ddd'],'module':['ABS,ABS,IPMA','BCCM','HPOC,ABS','ABS']})

Answer 1

检查下面的代码

df_merge = pd.merge(df2, df1.groupby(['vin'])['module'].apply(list), on ='vin', how ='left')
df_merge['module'] = df_merge['module'].astype('str').str.replace("\[|\]|\'| ","")
df_merge

输出：

Answer 2

您可以简单地做：

df2.merge(df1, how='left').groupby('vin').agg({'module': lambda x: ', '.join(x)})

它给您：

VIN	模块
AAA	ABS，ABS，IPMA
BBB	BCCM
CCC	HPOC，ABS
DDD	ABS
EEE	HPOC，ABS
FFF	ABS