Is there a way to convert list of values in Pandas DataFrame row to multiple columns?

Question

I am trying to convert a list of values in a Dataframe row into seperate columns

I have tried to use pivot,pivot_table and get_dummies but I couldnt get the expected result.

Actual Columns:

A      B

1  [1, 3, 5]
2  [2, 4, 6]

Expected Output:

 A B_1 B_2 B_3

 1  1   3   5
 2  2   4   6

df[['A']].join(pd.DataFrame(df.B.tolist(),columns=[f'B_{i+1}' for i in range(df.B.str.len().max())])) — anky
– anky, Commented Aug 31, 2019 at 18:03
Possible duplicate of Pandas split column of lists into multiple columns — Ehsan
– Ehsan, Commented Sep 1, 2019 at 18:20

Scott Boston · Accepted Answer · 2019-08-31 18:06:33Z

1

One way:

df_out = df.set_index('A').explode('B')
df_out = df_out.set_index(df_out.groupby(level=0).cumcount() + 1, append=True).unstack()
df_out.columns = [f'{i}_{j}' for i, j in df_out.columns]
df_out.reset_index()

Output:

   A B_1 B_2 B_3
0  1   1   3   5
1  2   2   4   6

answered Aug 31, 2019 at 18:06

Scott Boston

154k15 gold badges160 silver badges207 bronze badges

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

Collectives™ on Stack Overflow

Is there a way to convert list of values in Pandas DataFrame row to multiple columns?

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