Python 从分组数据框中获取列

在 Python 中，我们经常使用 pandas 库来处理和分析数据。当我们有一个包含分组数据的 DataFrame 时，我们可能需要从中提取特定列的数据。

下面是一些获取分组数据框中列的常用方法：

1. 使用列名称

我们可以使用 DataFrame 的列名称来获取其中的列数据。这适用于单个分组数据框。

# 导入 pandas 库
import pandas as pd

# 创建包含多个列的 DataFrame
data = {'A': [1, 1, 2, 2], 'B': ['a', 'b', 'c', 'd'], 'C': [True, False, True, False]}
df = pd.DataFrame(data)

# 根据列名称获取列数据
col_data = df['A']

2. 使用 iloc

我们可以使用 iloc 方法根据列的索引位置来获取列数据。这适用于单个或多个分组数据框。

# 获取第一列数据
col_data = df.iloc[:, 0]

# 获取多列数据
col_data = df.iloc[:, [0, 2]]

3. 使用 loc

我们可以使用 loc 方法根据列名称来获取列数据。这适用于单个或多个分组数据框。

# 获取单个列数据
col_data = df.loc[:, 'A']

# 获取多列数据
col_data = df.loc[:, ['A', 'B']]

4. 使用 get_group

如果我们有一个分组数据框，我们可以使用 get_group 方法根据组的名称获取特定组的数据。这适用于多个分组数据框。

# 根据组名称获取特定组的数据
group_data = df.get_group('Group A')

# 如果分组数据框是通过 groupby 方法创建的，我们可以这样获取特定组的数据：
grouped = df.groupby('Group')
group_data = grouped.get_group('Group A')

这些是从分组数据框中获取列的一些常用方法。希望这能帮助你在 Python 中处理数据时更方便地获取需要的列。