Python字典在CSV数据统计中的应用

精选原创

web安全工具库 2024-05-20 14:52:12 ©著作权

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

在数据分析和处理中，CSV（逗号分隔值）文件是一种常见的数据存储格式。Python提供了多种处理CSV文件的库，如csv和pandas。字典（Dictionary）是Python中一个非常有用的数据结构，它允许我们以键值对的形式存储数据。在处理CSV数据时，字典可以用来高效地统计和分析数据。本文将介绍如何使用Python字典来统计CSV数据，并提供几个实用的代码案例。

1. 读取CSV文件

首先，我们需要读取CSV文件。Python的csv模块提供了读取CSV文件的功能。以下是一个简单的例子，展示了如何使用csv模块读取CSV文件。

import csv

# 打开CSV文件
with open('data.csv', 'r') as file:
    reader = csv.reader(file)
    data = list(reader)

# 打印数据
for row in data:
    print(row)

2. 使用字典统计数据

假设我们有一个CSV文件，其中包含了一些商品的销售数据，每一行代表一次销售记录，包含商品名称和销售数量。我们可以使用字典来统计每个商品的总销售数量。

import csv

# 初始化字典
sales_dict = {}

# 打开CSV文件
with open('sales_data.csv', 'r') as file:
    reader = csv.reader(file)
    next(reader)  # 跳过标题行
    for row in reader:
        product = row[0]
        quantity = int(row[1])
        if product in sales_dict:
            sales_dict[product] += quantity
        else:
            sales_dict[product] = quantity

# 打印统计结果
for product, total_sales in sales_dict.items():
    print(f'Product: {product}, Total Sales: {total_sales}')

3. 代码案例：统计多个字段

有时候，我们需要统计CSV文件中多个字段的数据。例如，我们可能需要统计每个商品的平均销售价格。以下是一个代码案例，展示了如何使用字典来统计这些数据。

import csv

# 初始化字典
sales_dict = {}

# 打开CSV文件
with open('sales_data.csv', 'r') as file:
    reader = csv.reader(file)
    next(reader)  # 跳过标题行
    for row in reader:
        product = row[0]
        quantity = int(row[1])
        price = float(row[2])
        if product in sales_dict:
            sales_dict[product]['total_quantity'] += quantity
            sales_dict[product]['total_price'] += price * quantity
        else:
            sales_dict[product] = {'total_quantity': quantity, 'total_price': price * quantity}

# 计算平均价格
for product, data in sales_dict.items():
    total_quantity = data['total_quantity']
    total_price = data['total_price']
    average_price = total_price / total_quantity
    print(f'Product: {product}, Average Price: {average_price:.2f}')

4. 代码案例：使用`pandas`简化统计

pandas是一个强大的数据分析库，它提供了更简洁的方式来处理CSV文件。以下是一个代码案例，展示了如何使用pandas和字典来统计数据。

import pandas as pd

# 读取CSV文件
df = pd.read_csv('sales_data.csv')

# 使用groupby和字典统计数据
sales_dict = df.groupby('Product')['Quantity', 'Price'].sum().to_dict('index')

# 计算平均价格
for product, data in sales_dict.items():
    total_quantity = data['Quantity']
    total_price = data['Price']
    average_price = total_price / total_quantity
    print(f'Product: {product}, Average Price: {average_price:.2f}')