How to combine dataframe rows, and combine their string column into list?

Question

Say I have a Pandas dataframe:

index  name      A
0      one       a
1      two       a
2      one       b
3      two       a

How can I merge rows with identical 'name' so that the new column A is a list of all the A associated with each 'name'? So, the output would be:

index  name      A
0      one       [a, b]
1      two       [a]

Check the duplicated link, only need to add `drop_duplicate` before groupby. — Ynjxsjmh, May 24 '22 at 15:06

ArchAngelPwn · Accepted Answer · 2022-11-15T20:25:09.323

0

Update Answer

df = pd.DataFrame({
    'name' : ['one', 'two', 'one', 'two'],
    'A' : ['a', 'a', 'a', 'a'],
    'B': ['a', 'b', 'a', 'a']
})

df['COLUMN_MERGE'] = df['A'].astype(str) + ' ' + df['B'].astype(str)
df = df.groupby(['name'])['COLUMN_MERGE'].apply(set).apply(list).reset_index(name = 'Distinct_Merge')

edited Nov 15 '22 at 20:25

answered May 24 '22 at 15:08

ArchAngelPwn

2,891
1
4
17

what if we have more than 'A' column, how to merge them, please?? @ArchAngelPwn – M Khalil Nov 14 '22 at 17:49
I updated my answer, but this is a rough simplified example, if you wanted to get more in depth you would have to provide more details. – ArchAngelPwn Nov 15 '22 at 20:24

How to combine dataframe rows, and combine their string column into list?

1 Answers1