尝试将多个函数聚合到新列时意外的KeyError Pandas

时间:2014-11-22 11:20:15

标签: python pandas dictionary dataframe aggregate

我查看了以下问题:

Apply multiple functions to multiple groupby columns

我有

的数据
                    p.date p.instrument                p.sector  \
11372  2013-02-15 00:00:00            A             Health Care   
11373  2013-02-15 00:00:00           AA               Materials   
11374  2013-02-15 00:00:00         AAPL  Information Technology   
11375  2013-02-15 00:00:00         ABBV             Health Care   
11376  2013-02-15 00:00:00          ABC             Health Care   

                                p.industry    p.retn  p.pfwt     b.bwt  
11372     Health Care Equipment & Services -5.232929     NaN  0.000832  
11373                             Aluminum  0.328947     NaN  0.000907  
11374                    Computer Hardware -1.373927     NaN  0.031137  
11375                      Pharmaceuticals  2.756020     NaN  0.004738  
11376  Health Care Distribution & Services -0.371179     NaN  0.000859 

但是当我尝试时:

test1.groupby("p.sector").agg({'r1': lambda x: x['p.pfwt'].sum()})

我收到错误

KeyError: 'r1'

我尝试使用当前DataFrame的一组结果创建新列。

我错过了什么?感谢

1 个答案:

答案 0 :(得分:6)

使用

test1.groupby("p.sector").agg({'p.pfwt': np.sum})

例如,请参阅this pandas docs

  • 聚合字典中的键必须与数据帧中预先存在的键对应。您的程序失败,因为您的数据框中没有“r1”列,因此它无法聚合不存在的内容。
  • 如果您需要重命名结果,则可以为这样的系列添加链式操作:.agg([np.sum, np.mean, np.std]).rename(columns={'sum': 'foo', 'mean': 'bar', 'std': 'baz'}) )