熊猫是否可以执行包含两列的聚合操作？

Question

给定以下数据帧，
是否可以在单个聚合函数中计算col2和col2+col3的和，
？

import pandas as pd
df = pd.DataFrame({'col1': ['a', 'a', 'b', 'b'], 'col2': [1, 2, 3, 4], 'col3': [10, 20, 30, 40]})

在R的dplyr中，我会用单行的summary来完成，
我想知道在Pandas中有什么等价的：

df %>% group_by(col1) %>% summarize(col2_sum = sum(col2), col23_sum = sum(col2 + col3))

预期结果：

Answer 1

让我们先尝试assign新列

out = df.assign(col23 = df.col2+df.col3).groupby('col1',as_index=False).sum()

出[81]:

  col1  col2  col3  col23
0    a     3    30     33
1    b     7    70     77

根据我的理解，apply更像R中的summary

out = df.groupby('col1').\
           apply(lambda x : pd.Series({'col2_sum':x['col2'].sum(),
                                       'col23_sum':(x['col2'] + x['col3']).sum()})).\
               reset_index()
Out[83]: 
  col1  col2_sum  col23_sum
0    a         3         33
1    b         7         77