python计算某一列类别的个数和名称

原创

mob649e81607bf3 2023-12-09 06:22:25 ©著作权

文章标签 Python 数据分析读取数据 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob649e81607bf3的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python计算某一列类别的个数和名称

在数据分析和处理的过程中，我们经常需要统计某一列的类别个数和名称。Python提供了强大的工具来完成这个任务，本文将介绍如何使用Python计算某一列类别的个数和名称。

1. 数据准备

首先，我们需要准备一些数据来演示。假设我们有一个包含学生姓名、性别和年龄的表格数据，如下所示：

姓名	性别	年龄
张三	男性	20
李四	女性	22
王五	男性	20
赵六	女性	22
孙七	男性	21

我们的目标是计算性别列的类别个数和名称。

2. 使用Python计算类别个数和名称

Python提供了多种方式来计算某一列类别的个数和名称。下面我们将介绍两种常用的方法。

2.1 使用pandas库

pandas是Python中常用的数据分析库，提供了丰富的数据处理和统计函数。我们可以使用pandas来读取数据，然后使用value_counts函数来计算类别个数和名称。

import pandas as pd

# 读取数据
data = pd.read_csv('data.csv')

# 计算性别列的类别个数和名称
category_counts = data['性别'].value_counts()
category_names = category_counts.index.tolist()

print("类别个数：", len(category_names))
print("类别名称：", category_names)

运行以上代码，输出结果为：

类别个数： 2
类别名称： ['男性', '女性']

2.2 使用collections库

Python的collections库提供了Counter类，它是一种特殊的字典（dictionary），用于计数可哈希对象。我们可以使用Counter类来计算类别个数和名称。

from collections import Counter

# 读取数据
data = pd.read_csv('data.csv')

# 计算性别列的类别个数和名称
category_counts = Counter(data['性别'])
category_names = list(category_counts.keys())

print("类别个数：", len(category_names))
print("类别名称：", category_names)

运行以上代码，输出结果与上面的方法相同。