python - 将 kwargs 添加到带有字典参数的 pandas apply 语句

Question

我有以下 DataFrame，我试图从中通过字典参数获取多个聚合到 groupby().agg()

>>> apply_dict
{'Amount': ['sum', 'std'], 'TransactionDate': ['min']}
>>> df
  Account  Amount TransactionDate
0       A      10      10-20-2018
1       A      20      10-21-2018
>>> df.groupby('Account').agg(apply_dict)
        Amount           TransactionDate
           sum       std             min
Account                                 
A           30  7.071068      10-20-2018

但是，标准差被报告为样本标准差，但我想获得总体标准差。 使用字典时是否可以向 agg() 提供 kwarg？尝试以下调用会返回与上面相同的 DataFrame:

>>> df.groupby('Account').agg(apply_dict, ddof=0)
        Amount           TransactionDate
           sum       std             min
Account                                 
A           30  7.071068      10-20-2018

谢谢!

Answer 1

是的，只需通过 lambda 指定即可

apply_dict = {'Amount': ['sum', ('std', lambda s: s.std(ddof=0))], 
              'TransactionDate': ['min']}

        Amount     TransactionDate
           sum std             min
Account                           
A           30   5      10-20-2018