Pivot dataframe in Python

Question

I am trying to pivot a dataframe in Python for a school project. I know how to transform the rows into column, but what I don't know is how to populate that new columns with the value 1 or 0 depending the case. For example, now I have something like this:

Code   Product
1       Water
1       Coke
2       Wine
3       Apples

and I want to turn it into something like this:

Code   Water  Coke  Wine  Apples
1       1       1    0      0      
2       0       0    1      0
3       0       0    0      1

I searched the web and I could't find anything. Also, I'm not that advanced in Python language.

Thanks

yatu · Accepted Answer · 2019-03-21 14:15:52Z

2

You can do this with pd.crosstab:

pd.crosstab(index=df.Code, columns=df.Product)

Product   Apples  Coke  Water  Wine
Code                              
1             0     1      1     0
2             0     0      0     1
3             1     0      0     0

answered Mar 21, 2019 at 14:15

yatu

88.6k12 gold badges93 silver badges148 bronze badges

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

2 Comments

fazistinho_ Over a year ago

How would you do the opposite! Turn a df that looked like the crosstab into a 1 row/column df?

yatu Over a year ago

Check pd.melt @HiFizzle_

Collectives™ on Stack Overflow

Pivot dataframe in Python

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related