我想在下面的代码中只有一个值列:
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
但是,我想要的是一个新名称的专栏:
dqratio
team
a 0.029126
答案 0 :(得分:1)
我认为您正在寻找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