Group by Pandas and combine multipe strings

Question

I have a dataframe currently which resembles this

Source | Destination | Type
A | B | Insert
A | B | Delete
B | C | Insert

What I want to achieve is something like this

Source | Destination | Type
A | B | Insert, Delete
B | C | Insert

I tried using group by Source and Destination but Im a little unsure how do I append to the type. Any ideas?

Does this answer your question? [How to combine multiple rows into a single row with pandas](https://stackoverflow.com/questions/36392735/how-to-combine-multiple-rows-into-a-single-row-with-pandas) — Vladimir Fokow, Aug 14 '22 at 14:08

score 0 · Answer 1 · answered Aug 14 '22 at 14:04

0

Check Below Code:

df.groupby(['Source','Destination']).agg({'Type':'unique'})

answered Aug 14 '22 at 14:04

Abhishek

1,585
2
12
15

score 0 · Answer 2 · answered Aug 14 '22 at 14:05

0

Figured out something

df = df.groupby(['Source','Destination'])['Type'].apply(lambda x: ','.join(x)).reset_index()

answered Aug 14 '22 at 14:05

Adam

1,157
6
19
40

score 0 · Answer 3 · answered Aug 14 '22 at 14:06

0

Make a string separated by ", ":

df.groupby(['Source', 'Destination']).agg(', '.join).reset_index()

Make a list:

df.groupby(['Source','Destination']).agg(list).reset_index()

answered Aug 14 '22 at 14:06

Vladimir Fokow

3,728
2
5
27

score 0 · Answer 4 · answered Aug 14 '22 at 14:45

Joining by , and building a string works but it is not very convenient if you later want to perform other operations with this column value. (such as iterating over the elements.)

This creates a set of values.

column_map: Dict[str,Any] = {} column_map["Type"] = lambda x: set(x)

df.groupby(["Source", "Destination"]).agg(column_map)

Source | Destination | Type
A | B | {Insert Delete}
B | C | {Insert}

if you instead want to get a list and dont want eliminate duplicates. Just replace set(x) with list(x)

Group by Pandas and combine multipe strings

4 Answers4