pandas aggregated data to a numpy array : data structure conversion

Question

I have aggregated data using pandas data frame. Below is some actual data shown and how I aggregated it.

fdf.groupby(['row',col'])['percent'].sum()

What I would like to do is create a 2d numpy array of this (rows = row, columns = col). Any slick way to do this ?

Another way I did something similar was create a pivot table

pivot_table(fdf,values='percent',rows='row',cols='col', aggfunc=np.sum)

In this case I want to convert this pivot table to 2d numpy array. Is there a way for me to index into each cell of this table. If so then I probably will be Ok with the table itself.

Wes McKinney · Accepted Answer · 2012-04-18 21:50:13Z

6

Try:

result = fdf.groupby(['row',col'])['percent'].sum()
result.unstack('col').values

Alternately:

fdf.pivot_table('percent', rows='row', cols='col', aggfunc='sum').values

answered Apr 18, 2012 at 21:50

Wes McKinney

106k32 gold badges146 silver badges109 bronze badges

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

Collectives™ on Stack Overflow

pandas aggregated data to a numpy array : data structure conversion

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