如何從Pandas dataFrame返回groupby值？

Question

我有一個dataFrame我總結了一些數據，我正在繪制它。 我想從groupby字段中獲取值以用作圖中的標簽。

這是我的匯總dataFrame：

code_values = code_graph.groupby(['code_desc']).agg({'product_id': pd.Series.nunique })

哪個回報：

print code_values
                          product_id
code_desc                          
Product 1                         30
Product 2                         40
Product 3                         50
Product 4                         60

當我嘗試訪問code_desc字段以在matplotlib圖中用作我得到的標簽時：

print code_values['code_desc']
KeyError: u'no item named code_desc'

如何獲取code_desc的值？

Answer 1

從聚合值，您應該能夠使用：

code_values.index

您還可以獲取groupby對象的組字典的鍵，如：

grouped = code_graph.groupby(['code_desc'])
code_desc = grouped.groups.keys()

Answer 2

如果您想以值的形式訪問它們，它們可能不再是您的索引。 最后考慮一下

code_values.reset_index(inplace=True)
print code_values['code_desc']

但是，許多大熊貓的內置可視化工具實際上會根據您的索引繪制值。 那么，也許你想保留索引，但看看這些？