如何将数据行分组到 Pandas Groupby 中的列表中？

Pandas 中的 groupby 函数是一种将数据分组并对每个组应用函数的强大工具。本篇文章将教你如何使用 groupby 函数将数据行分组到 Pandas Groupby 中的列表中。

准备工作

在开始之前，我们需要准备一个数据集。本文将使用以下数据集：

import pandas as pd

data = {
    '类别': ['水果', '水果', '蔬菜', '蔬菜', '肉类', '肉类', '其他'],
    '销售额': [100, 120, 200, 150, 300, 250, 50]
}

df = pd.DataFrame(data)

分组数据

我们使用 groupby 函数将数据分组。我们现在将按 类别 列对数据进行分组。代码如下：

grouped_data = df.groupby('类别')

这将返回一个 pandas.core.groupby.generic.DataFrameGroupBy 对象。我们可以在此对象上应用各种操作来处理分组数据。

在分组中创建列表

在本例中，我们将对分组中的销售额创建列表。我们使用 apply 函数将一些 lambda 函数应用于销售额列。在 lambda 函数中，我们使用 tolist 函数将 Series 对象转换为列表。

以下是完整代码：

grouped_data = df.groupby('类别')
sales_list = grouped_data['销售额'].apply(lambda x: x.tolist())

现在，sales_list 变量将包含每个类别的销售额列表。

结论

使用 Pandas 的 groupby 函数，我们可以轻松地将数据行分组到 Pandas Groupby 中的列表中。我们可以对分组后的数据执行各种操作，创建各种统计和汇总信息。