datalib icon indicating copy to clipboard operation
datalib copied to clipboard

How to get equivalent of pandas.DataFrame.pivot_table?

Open pikhovkin opened this issue 6 years ago • 0 comments

How to get equivalent of pandas.DataFrame.pivot_table like here?

table = pivot_table(df, values='D', index=['A', 'B'], columns=['C'], aggfunc=[np.sum])

Input:

>>> df
     A    B      C  D
0  foo  one  small  1
1  foo  one  large  2
2  foo  one  large  2
3  foo  two  small  3
4  foo  two  small  3
5  bar  one  large  4
6  bar  one  small  5
7  bar  two  small  6
8  bar  two  large  7

Output:

C        large  small
A   B
bar one    4.0    5.0
    two    7.0    6.0
foo one    4.0    1.0
    two    NaN    6.0

pikhovkin avatar May 21 '18 14:05 pikhovkin