当前位置: 代码网 > it编程>前端脚本>Python > Python用字典统计CSV数据的实现示例

Python用字典统计CSV数据的实现示例

2024年06月13日 Python 我要评论
在数据分析过程中,csv(comma-separated values)文件是一种常见的数据存储格式。python 提供了许多处理csv文件的工具,其中字典(dictionary)是一个非常高效的数据

在数据分析过程中,csv(comma-separated values)文件是一种常见的数据存储格式。python 提供了许多处理csv文件的工具,其中字典(dictionary)是一个非常高效的数据结构,可以用来统计和处理csv数据。本文将详细介绍如何使用python字典统计csv数据,并通过多个实例来说明其应用场景。

一、准备工作

在开始之前,我们需要安装和导入必要的库。这里我们使用python内置的csv模块来读取csv文件,同时使用collections模块中的defaultdict来方便地统计数据。

import csv
from collections import defaultdict

二、读取csv文件

首先,让我们看看如何读取一个csv文件。假设我们有一个名为data.csv的文件,内容如下:

name,age,city
alice,30,new york
bob,25,los angeles
alice,32,new york
david,25,chicago

我们可以使用以下代码来读取文件:

filename = 'data.csv'

with open(filename, mode='r', newline='') as file:
    reader = csv.dictreader(file)
    data = [row for row in reader]

print(data)

运行上述代码会输出:

[{'name': 'alice', 'age': '30', 'city': 'new york'},
 {'name': 'bob', 'age': '25', 'city': 'los angeles'},
 {'name': 'alice', 'age': '32', 'city': 'new york'},
 {'name': 'david', 'age': '25', 'city': 'chicago'}]

三、使用字典统计数据

统计每个城市的人员数量

我们可以使用字典来统计每个城市的人员数量:

city_count = defaultdict(int)

for row in data:
    city = row['city']
    city_count[city] += 1

print(dict(city_count))

运行结果:

{'new york': 2, 'los angeles': 1, 'chicago': 1}

统计每个名字出现的次数

类似地,我们可以统计每个名字出现的次数:

name_count = defaultdict(int)

for row in data:
    name = row['name']
    name_count[name] += 1

print(dict(name_count))

运行结果:

{'alice': 2, 'bob': 1, 'david': 1}

统计每个城市的平均年龄

我们还可以统计每个城市的平均年龄。这需要两个步骤:首先统计每个城市的总年龄和人数,然后计算平均年龄。

city_age_sum = defaultdict(int)
city_count = defaultdict(int)

for row in data:
    city = row['city']
    age = int(row['age'])
    city_age_sum[city] += age
    city_count[city] += 1

city_avg_age = {city: city_age_sum[city] / city_count[city] for city in city_age_sum}

print(city_avg_age)

运行结果:

{'new york': 31.0, 'los angeles': 25.0, 'chicago': 25.0}

四、多场景应用

除了上述基础统计,字典还可以用于更复杂的数据处理场景。下面是几个应用实例:

统计每个名字在每个城市的出现次数

name_city_count = defaultdict(lambda: defaultdict(int))

for row in data:
    name = row['name']
    city = row['city']
    name_city_count[name][city] += 1

print({name: dict(city_count) for name, city_count in name_city_count.items()})

运行结果:

{'alice': {'new york': 2},
 'bob': {'los angeles': 1},
 'david': {'chicago': 1}}

统计每个年龄段的人数

假设我们要统计20-29岁、30-39岁等年龄段的人数:

age_group_count = defaultdict(int)

for row in data:
    age = int(row['age'])
    if 20 <= age < 30:
        age_group = '20-29'
    elif 30 <= age < 40:
        age_group = '30-39'
    # 其他年龄段可以继续添加
    else:
        age_group = 'other'
    
    age_group_count[age_group] += 1

print(dict(age_group_count))

运行结果:

{'30-39': 2, '20-29': 2}

五、总结

本文介绍了如何使用python字典统计csv数据,并通过多个实例展示了其在不同场景下的应用。字典作为一种灵活高效的数据结构,可以极大地简化数据统计和处理的过程。希望这些示例能够帮助你更好地理解和应用python字典来处理实际数据分析任务。

到此这篇关于python用字典统计csv数据的实现示例的文章就介绍到这了,更多相关python 字典统计csv内容请搜索代码网以前的文章或继续浏览下面的相关文章希望大家以后多多支持代码网!

(0)

相关文章:

版权声明:本文内容由互联网用户贡献,该文观点仅代表作者本人。本站仅提供信息存储服务,不拥有所有权,不承担相关法律责任。 如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 2386932994@qq.com 举报,一经查实将立刻删除。

发表评论

验证码:
Copyright © 2017-2025  代码网 保留所有权利. 粤ICP备2024248653号
站长QQ:2386932994 | 联系邮箱:2386932994@qq.com