CODEWITHSUNDEEP
pythonpython-2.7pandasdataframetranspose
Aidis
Aidis

Reputation: 1280

Pandas transposition inside dataframe

I have this datasate now:

animal   age             count
dogs     1               49
         2              134
         3              147
         4              154
cats     1              189
         2              254
         3              259
         4              261

I would like to convert age column to 4 age columns for each age:

animal   age1 age2 age3 age4 
dogs      49   134  147  154  
cats     ....................

I have tried df.T and df.transpose() but both of them return my original column.

Upvotes: 1

Views: 5523

Answers (1)

unutbu
unutbu

Reputation: 879591

You could use pd.pivot:

In [25]: result = df.pivot(index='animal', columns='age', values='count')

In [26]: result
Out[26]: 
age       1    2    3    4
animal                    
cats    189  254  259  261
dogs     49  134  147  154

In [27]: result.columns = ['age{:d}'.format(col) for col in result.columns]

In [28]: result
Out[28]: 
        age1  age2  age3  age4
animal                        
cats     189   254   259   261
dogs      49   134   147   154

Upvotes: 7

Related Questions