How to expand pandas column containing values in arrays to multiple columns?

Question

I have a data frame with column named 'gear' which contains lists of values... what i want to do now is to move each element in every list to the corresponding column.

is there a way to do this without a need for a for loop? for example in the first row the value 'Hengerfeste' in the list should be moved to 'Hengerfeste' column and so on for each element in the list.

it takes very long main problem with that are the multiple calls to .iloc to assign the values — Sunny
– Sunny, Commented Oct 13, 2021 at 13:50

Quang Hoang · Accepted Answer · 2021-10-13 13:51:55Z

2

Try explode, then groupby().value_counts():

#sample data
df = pd.DataFrame({'col':[['a','b','c'], ['a','c','x'],[],['b','x','y']]})


(df['your_list_col'].explode()
   .groupby(level=0).value_counts()
   .unstack(fill_value=0)
   .reindex(df.index, fill_value=0)
)

Output:

col  a  b  c  x  y
0    1  1  1  0  0
1    1  0  1  1  0
2    0  0  0  0  0
3    0  1  0  1  1

answered Oct 13, 2021 at 13:51

Quang Hoang

151k11 gold badges64 silver badges86 bronze badges

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

1 Comment

Sunny Over a year ago

man you are a genius! thanks this works!

Collectives™ on Stack Overflow

How to expand pandas column containing values in arrays to multiple columns?

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related