Convert the lists in multiple columns to rows in pandas df

Question

I have df with three columns:

Col 0 - sentence row num
Col 1 - Sentence converted to list
Col 2 - list of annotations

Col 0	Col1	Col2
1	[This, is, sentence]	[l1, l2, l3]
2	[This, is, sentence, too]	[l1, l2, l3, l4]

I would like to convert Col1 and Col2 and move each row and its respective annotation to separate row:

Col 0	Col1	Col2
1	This	l1
1	is	l2
1	sentence	l3
2	This	l1
2	is	l2
2	sentence	l3
2	too	l4

When I use explode on each column separately one of the columns always does not change.

data2['Col1_exploded'] = (data['Col1'].explode('Col1'))

And this option does not work too:

data2[['Col1_exploded', 'Col2_exploded']] = (data[['Col1', 'Col2']].explode('Col1', 'Col2'))

ThePyGuy · Accepted Answer · 2022-10-06 13:11:35Z

2

You can pass list of column names to explode:

>>> df.explode(['Col1', 'Col2'])

   Col 0      Col1 Col2
0      1      This   l1
0      1        is   l2
0      1  sentence   l3
1      2      This   l1
1      2        is   l2
1      2  sentence   l3
1      2       too   l4

answered Oct 6, 2022 at 13:11

ThePyGuy

18.5k5 gold badges24 silver badges55 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

grymlin · Accepted Answer · 2022-10-06 13:14:21Z

1

you were really close ! this works for me :

df.explode(['Col1', 'Col2'])

result :

    Col 0   Col1    Col2
0   1   This    l1
0   1   is  l2
0   1   sentence    l3
1   2   This    l1
1   2   is  l2
1   2   sentence    l3
1   2   too l4

answered Oct 6, 2022 at 13:14

grymlin

5022 silver badges10 bronze badges

Collectives™ on Stack Overflow

Convert the lists in multiple columns to rows in pandas df

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related