exploding list-columns in pandas

Question

Consider the following example

dftest = pd.DataFrame({'mylist1' : [['hello', 'hasta'], 'one'],
                       'mylist2' : [['there', 'la vista'], 'shot']})

dftest
Out[240]: 
          mylist1            mylist2
0  [hello, hasta]  [there, la vista]
1             one               shot

I would like to explode the two columns so that the nth element in mylist1 is concatenated to the nth element in mylist2. Mylist1 and mylist2 always have the same number of elements (in this example: 2 in the first obs, 1 in the second obs).

The desired output is shown below. As you can see hello is matched with there, hasta is matched with la vista, etc. We obtain three rows because there are two elements in the first list, and just one in the second one.

Out[241]: 
         exploded
0     hello there
1  hasta la vista
2        one shot

How can I do this? Thanks!

Nk03 · Accepted Answer · 2021-07-06 17:48:16Z

1

Here's one way:

Explode the dataframe vertically
join strings along axis 1.

df = df.apply(pd.Series.explode).apply(' '.join, 1)

OUTPUT:

0       hello there
0    hasta la vista
1          one shot
dtype: object

edited Jul 6, 2021 at 17:48

answered Jul 6, 2021 at 17:42

Nk03

15k2 gold badges11 silver badges24 bronze badges

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

Collectives™ on Stack Overflow

exploding list-columns in pandas

1 Answer 1

OUTPUT:

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

OUTPUT:

Comments

Your Answer

Sign up or log in

Post as a guest

Related