python统计csv文件某列不同值的个数

原创

mob649e816a3664 2023-12-04 14:52:33 ©著作权

文章标签 CSV 代码示例 Python 文章分类 Python 后端开发

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

Python统计CSV文件某列不同值的个数

CSV（Comma-Separated Values）是一种常用的文件格式，通常用于存储和交换数据。在数据分析和处理中，经常需要统计CSV文件中某列的不同值的个数，这可以帮助我们对数据的分布和特征有更深入的了解。

本文将介绍如何使用Python统计CSV文件中某列的不同值的个数，同时提供代码示例和相关的科普知识。

1. CSV文件的读取

在Python中，我们可以使用csv模块来读取和处理CSV文件。首先，我们需要使用open()函数打开CSV文件，并使用csv.reader()方法创建一个CSV读取器对象。通过遍历读取器对象，我们可以逐行读取CSV文件中的数据。

下面是一个读取CSV文件的代码示例：

import csv

filename = 'data.csv'

with open(filename, 'r') as csvfile:
    reader = csv.reader(csvfile)
    for row in reader:
        print(row)  # 打印每一行数据

2. 统计某列的不同值的个数

在读取CSV文件的基础上，我们可以使用Python的集合（Set）来统计某一列的不同值的个数。集合是一种无序且不重复的数据结构，非常适合用于计算不同元素的个数。

下面是一个统计某列不同值的个数的代码示例：

import csv

filename = 'data.csv'

values = set()  # 创建一个空的集合

with open(filename, 'r') as csvfile:
    reader = csv.reader(csvfile)
    next(reader)  # 跳过表头
    for row in reader:
        values.add(row[2])  # 将每一行的第三列元素添加到集合中

count = len(values)  # 统计集合中不同元素的个数
print("不同值的个数：", count)

在上述代码中，我们使用了一个set来存储CSV文件中某一列的不同值。通过遍历CSV文件中的每一行，我们将每一行的第三列元素添加到集合中。最后，通过len()函数计算集合的长度，即不同值的个数。

3. 示例应用

以上是一个简单的代码示例，展示了如何统计CSV文件中某列不同值的个数。实际应用中，我们可以基于这个方法进行更复杂的数据分析和处理。

例如，假设我们有一个CSV文件记录了用户的购买记录，其中一列是商品名称。我们可以使用上述方法统计不同商品的个数，进而分析用户购买偏好和热门商品。

import csv

filename = 'purchase_records.csv'

products = set()  # 创建一个空的集合

with open(filename, 'r') as csvfile:
    reader = csv.reader(csvfile)
    next(reader)  # 跳过表头
    for row in reader:
        products.add(row[1])  # 将每一行的第二列元素（商品名称）添加到集合中

count = len(products)  # 统计集合中不同商品的个数
print("不同商品的个数：", count)