pandas groupby后提取一组

当你使用 Pandas 的 groupby 函数进行分组后，可以通过 get_group 方法获取某一组数据。get_group 方法需要传入一个参数，即你需要提取的组的名称，通常是组的值。

下面是一个示例代码，假设我们有一个DataFrame df，其中有两列 A 和 B，现在我们要按照 A 列进行分组并获取其中一个组的数据：

import pandas as pd

df = pd.DataFrame({'A': ['foo', 'bar', 'foo', 'bar', 'foo', 'bar', 'foo', 'foo'],
                   'B': ['one', 'one', 'two', 'three', 'two', 'two', 'one', 'three'],
                   'C': [1, 2, 3, 4, 5, 6, 7, 8],
                   'D': [9, 10, 11, 12, 13, 14, 15, 16]})

grouped = df.groupby('A')

# 获取 A 列为 foo 的一组数据
foo_group = grouped.get_group('foo')

print(foo_group)

输出结果：

     A    B  C   D
0  foo  one  1   9
2  foo  two  3  11
4  foo  two  5  13
6  foo  one  7  15
7  foo  three 8 16