python - 按数据框中的索引和列聚合数据

Question

我有这样的股价数据。日期实际上是数据框的索引。其余的都是列。

date         open    high    low    close  name
2013-01-01   14.50   14.80   14.25  14.70  AAA
2013-01-02   14.70   14.90   14.30  14.68  AAA
2013-01-01   14.50   14.80   14.25  14.70  BBB
2013-01-02   14.70   14.90   14.30  14.68  BBB
2013-01-01   14.50   14.80   14.25  14.70  CCC
2013-01-02   14.70   14.90   14.30  14.68  CCC

我想要一个新的数据框，其中包含每个唯一名称的每月开盘价、最高价、最低价和收盘价的平均值。

我能得到的最好的结果是使用此代码，但没有“名称”列。

stock_agg = {
  'open': {'ave_open': 'mean'},
  'high': {'ave_high': 'mean'},
  'low': {'ave_low': 'mean'},
  'close': {'ave_close': 'mean'}
}
stock_group = stock_data.groupby([pd.Grouper(freq='M')]).agg(stock_agg)

在上面的示例中如何按名称和月份进行分组？

Answer 1

pandas.DataFrame.groupby 可以处理多个by:

import pandas as pd

df.groupby([df.index.month, 'name']).mean()

输出:

           open   high     low  close
date name                            
1    AAA   14.6  14.85  14.275  14.69
     BBB   14.6  14.85  14.275  14.69
     CCC   14.6  14.85  14.275  14.69