如何通过分组索引访问 pandas groupby 数据框?

ast*_*erd 4 python python-3.x pandas pandas-groupby

按照这个例子,我可以创建一个简单的数据框和groupby

import pandas as pd

# Create a sample data frame
df = pd.DataFrame({'A': ['foo', 'foo', 'foo', 'bar', 'bar'],
                   'B': range(5), 'C': range(5)})

# group by 'A' and sum 'B'
gf = df.groupby('A').agg({'B': 'sum'})
Run Code Online (Sandbox Code Playgroud)

结果是分组数据框 gf

    B
A   
bar 7
foo 3
Run Code Online (Sandbox Code Playgroud)

我想通过分组索引访问 gf 。就像是...

gf['foo'] returns 3 
gf['bar'] returns 7
Run Code Online (Sandbox Code Playgroud)

我还想按分组索引进行绘图。就像是...

gf.plot('A', 'B') such that  x=['foo','bar'], y=[3,7]
Run Code Online (Sandbox Code Playgroud)

LN_*_*N_P 6

gf.reset_index(level=0, inplace=True)

gf[gf.A == 'bar']
Run Code Online (Sandbox Code Playgroud)

返回:

     A  B
0  bar  7
Run Code Online (Sandbox Code Playgroud)

阴谋:

import matplotlib.pyplot as plt

plt.bar(gf.A, gf.B)
Run Code Online (Sandbox Code Playgroud)