使用Pandas groupby().ngroups 获取分组数量

在使用Pandas进行数据分析时，groupby() 函数是一个非常强大的工具，它可以根据指定的列对DataFrame进行分组。 当你想知道数据被分成了多少个组时，可以使用 ngroups 属性。

**代码示例：**pythonimport pandas as pd

创建示例DataFramedf = pd.DataFrame({ 'School': ['A', 'A', 'B', 'B', 'B'], 'Class': ['1', '2', '1', '2', '3'], 'Student': ['Tom', 'Jerry', 'Spike', 'Tyke', 'Butch']})

按 'School' 和 'Class' 分组grouped = df.groupby(['School', 'Class'])

获取分组数量num_groups = grouped.ngroups

打印结果print(f'分组数量：{num_groups}')

输出：

分组数量：5

解释：

1. 我们首先创建一个示例DataFrame，其中包含 'School'、'Class' 和 'Student' 三列。2. 使用 df.groupby(['School', 'Class']) 按 'School' 和 'Class' 列进行分组。3. grouped.ngroups 返回分组后的组数，即5。

总结:

groupby().ngroups 提供了一种快速简便的方法来获取DataFrame分组后的组数，这在数据分析中非常有用。