总结一组在与大熊猫乘以常数

Question

我有一个数据帧df

id value text route distance 
1  100 asf  a1 10 
2  200 sdf  a1 30 
3  345 efs  a2 40 
4  232 asd  a2 60 
5  565 yjy  a2 70 

我正在寻找的是由路由分组和总结距离栏和值列，但增加+ 20 *计数（ ID）为每个路线。

即

for route a1 sum(Value) is 300 + 20*(count of id in group a1) 

我最后的DF应该像

route value distance 
a1  340  40  #300+(2*20) since count is 2 in group a1 
a2  1202  170 #1142+(3*20) since count is 2 in group a2 

我可以使用聚合得到正常组和总和，

x=final_df_results.groupby(['route']).agg({'distance':sum,'value': sum}) 

，但它不会帮我

Answer 1

0）由于ID计数将在相同的value计数，您可以使用自定义lambda函数

In [175]: (df.groupby('route',as_index=False) 
      .agg({'distance': 'sum', 'value': lambda x: x.sum() + len(x)*20})) 
Out[175]: 
    route distance value 
0 a1  40 340 
1 a2  170 1202 

1）使用assign你可以写一个伪一行代码

In [166]: (df.groupby('route', as_index=False) 
      .agg({'distance': 'sum', 'value': 'sum', 'id': 'size'}) 
      .assign(value=lambda x: x.value + 20*x.id) 
      .drop('id', 1)) 
Out[166]: 
    route distance value 
0 a1  40 340 
1 a2  170 1202 

---

2）或者，分配中间输出

个

In [167]: dff = (df.groupby('route', as_index=False) 
        .agg({'distance': 'sum', 'value': 'sum', 'id': 'size'})) 

In [169]: dff['value'] = dff['value'] + 20*dff['id'] 

In [170]: dff 
Out[170]: 
    route distance id value 
0 a1  40 2 340 
1 a2  170 3 1202 

降id列，如果你需要

In [171]: dff.drop('id', axis=1) 
Out[171]: 
    route distance value 
0 a1  40 340 
1 a2  170 1202