python数据透视表对行数统计_python – 如何在数据透视表上添加总和和计数

这是我的数据集

customer_id hour size

1 0 1

1 1 18

2 1 7

这是我的代码

table = a.pivot_table(index=['customer_id'],

columns='hour',

fill_value=0,

values='size')

这就是我所拥有的

hour 0 1

customer_id

1 1 18

2 8 7

我需要的

hour 0 1 count sum

customer_id

1 1 18 2 19

2 0 7 1 7

count 1 2

sum 1 25

count是类别中的非零计数,sum是类别中的总和

解决方法:

一个可能的动态解决方案是省略fill_value = 0：

table = a.pivot_table(index='customer_id',

columns='hour',

values='size')

print (table)

hour 0 1

customer_id

1 1.0 18.0

2 NaN 7.0

a = table.agg(['count','sum'])

b = table.T.agg(['count','sum']).T

print (table.fillna(0).append(a).join(b))

0 1 count sum

1 1.0 18.0 2.0 19.0

2 0.0 7.0 1.0 7.0

count 1.0 2.0 NaN NaN

sum 1.0 25.0 NaN NaN

标签：python,dataframe,pandas,pivot-table