定义带有两列的aggfunc作为 pandas 数据透视表中的参数 [英] define aggfunc with two columns as arguments in pandas pivot table

问题描述

在下面的代码中，我只想要一个值列:

I want only one value column as a result in below code:

df = pd.DataFrame({'team':['a','a'],'balance':[100,3],'dpd':[0,60]})
df.pivot_table(index='team',values=['balance','dpd'], 
               aggfunc=lambda x: np.sum(np.where(x.dpd>=30,x.balance,0))/np.sum(x.balance))

此返回:

      balance      dpd
team 
a    0.029126 0.029126

但是，我想要的是一列新名称:

But, what I want is a column with new name :

        dqratio
team
a       0.029126

推荐答案

我认为您正在寻找groupby和apply

df.groupby('team').apply(lambda x: np.sum(np.where(x['dpd']>=30,x['balance'],0))/np.sum(x['balance'])).to_frame('dqratio')
       dqratio
team          
a     0.029126

这篇关于定义带有两列的aggfunc作为 pandas 数据透视表中的参数的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！