Pandas change dataframe structure

Question

Im currently following Wes McKinneys tutorials on pandas and I was wondering if it is possible to do a certain thing in it. I'll explain this with an example: I have a Data Frame :

    A   B   C
0   1 foo   data1
1   5 foo2  data2
2   8 foo3  data3
3   6 foo   data4
4   5 foo3  data5
5   3 foo2  data1

I want to transform the previous data frame to a new one that looks like this:

      data1 data2 data3 data4 data5
foo    1      0     0     6     0
foo2   3      5     0     0     0
foo3   0      0     8     0     5

Also, I can't have the same foo-data combo twice, so I dont care about overwriting values.

Is this possible with pandas?

Fabio Lamanna · Accepted Answer · 2015-12-10 13:54:02Z

5

You can pivot your original dataframe df setting the correct values for index, columns and values.

a = df.pivot_table(index='B',columns='C',values='A').fillna(0)

The fillna(0) replace NaN with 0 values.

C     data1  data2  data3  data4  data5
B                                      
foo       1      0      0      6      0
foo2      3      5      0      0      0
foo3      0      0      8      0      5

answered Dec 10, 2015 at 13:54

Fabio Lamanna

21.7k24 gold badges95 silver badges126 bronze badges

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

Collectives™ on Stack Overflow

Pandas change dataframe structure

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related