如何在 Python 中使用 group by 获取计数

在数据分析和数据处理中，我们经常需要使用 group by 对数据进行聚合操作。其中，在得到分组后的数据时，我们还需要统计各个分组的数量（即计数）。本文将介绍如何在 Python 中使用 group by 获取计数。

1. 准备数据

我们先准备一份示例数据。假设我们有一份销售数据，其中包括了产品类型（type）和销售金额（amount）两个字段。我们要对该数据进行分析，以便统计每种产品类型的销售数量。

import pandas as pd

# 准备数据
data = {'type': ['clothes', 'clothes', 'shoes', 'shoes', 'bags', 'bags'],
        'amount': [100, 200, 150, 50, 120, 80]}
df = pd.DataFrame(data)
print(df)

输出如下：

      type  amount
0  clothes     100
1  clothes     200
2    shoes     150
3    shoes      50
4     bags     120
5     bags      80

2. 使用 group by 获取计数

接下来，我们使用 group by 对产品类型进行分组，并统计每个分组的数量：

# 使用 group by 对产品类型进行分组，并统计每个分组的数量
result = df.groupby('type').size().reset_index(name='count')
print(result)

输出如下：

      type  count
0     bags      2
1  clothes      2
2    shoes      2

如上所示，我们使用了 groupby 方法对数据进行了分组操作，并且使用 size 方法获取了每个分组的数量。其中，reset_index 方法用于重置索引，并为计数值添加了一个名称为 count 的列。

3. 结论

本文介绍了如何在 Python 中使用 group by 获取计数的方法。在使用时，我们只需要先对数据进行分组，然后使用 size 方法获取每个分组的数量即可。这个方法适用于各种数据类型（如列表、Pandas DataFrame 等）和分组情况，具有很高的灵活性。