Pandas split multiple columns of lists

Question

A similar question has been answered here Pandas split column of lists into multiple columns

However, the solution can only apply to a single column of list, how about if I have multiple columns of lists, is there any better ways to splitting those columns instead of using the solution above to break down the column one by one?

For example

df = pd.DataFrame({"teams": [["SF", "NYG"] for _ in range(7)], "cities": [["NY", "LA"] for _ in range(7)]})

    teams       cities
0   [SF, NYG]   [NY, LA]
1   [SF, NYG]   [NY, LA]
2   [SF, NYG]   [NY, LA]
3   [SF, NYG]   [NY, LA]
4   [SF, NYG]   [NY, LA]
5   [SF, NYG]   [NY, LA]
6   [SF, NYG]   [NY, LA]

to

  team1 team2 city1 city2
0    SF   NYG    NY   LA
1    SF   NYG    NY   LA
2    SF   NYG    NY   LA
3    SF   NYG    NY   LA
4    SF   NYG    NY   LA
5    SF   NYG    NY   LA
6    SF   NYG    NY   LA

Thanks in advance

Ferris · Accepted Answer · 2021-06-30 03:58:06Z

1

I found a solution while writing this question,

splits = [pd.DataFrame(df[col].tolist()).add_prefix(col) for col in df.columns]
clean_df = pd.concat(splits, axis=1)

edited Jun 30, 2021 at 3:58

Ferris

5,6611 gold badge18 silver badges27 bronze badges

answered Jun 30, 2021 at 2:03

Dy1012

337 bronze badges

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

Collectives™ on Stack Overflow

Pandas split multiple columns of lists

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related