python - 在熊猫中检查相同的行并有条件地创建新列

Question

我有这样的数据框

df = pd.DataFrame({'User':['101','101','101','102','102','101','101','102','102','102'],'Country':['India','Japan','India','Brazil','Japan','UK','Austria','Japan','Singapore','UK'],
               'Name':['RN','TN','AP','AP','TN','TN','TS','RN','TN','AP']})

如果用户和国家/地区相同，我想在其他列中组合名称列值，如下所示

标签： pythonpython-3.xpandas

Answer 1

你可以让

df['Name_E'] = df.groupby(['User', 'Country']).Name.transform(lambda x: str.join(', ', x))