Python DictReader读写csv文件

最新推荐文章于 2025-02-06 14:57:22 发布

原创最新推荐文章于 2025-02-06 14:57:22 发布 · 4.2w 阅读

77 ·

CC 4.0 BY-SA版权

文章标签：

#python #csv #dictreader #读写csv

Python 专栏收录该内容

2 篇文章

订阅专栏

本文介绍如何使用Python的csv库进行CSV文件的读取与写入操作。通过示例代码展示了如何利用DictReader按行读取数据，并利用DictWriter写入包含列标题的数据。

Python3.8

Conda

Python

Python 是一种高级、解释型、通用的编程语言，以其简洁易读的语法而闻名，适用于广泛的应用，包括Web开发、数据分析、人工智能和自动化脚本

准备

此文档的示例代码是基于python3.5写的。
使用csv库前，先导入csv库：
import csv

读取内容

假设csv文件的内容如下图所示，DictReader会将第一行的内容（类标题）作为key值，第二行开始才是数据内容。即图中的csv文件有2列7行数据，第一列的key值为id，第二列的key值为class：
这里写图片描述

with open(‘name.csv’) as csvfile:
    reader = csv.DictReader(csvfile)
    for row in reader:
        #循环打印数据的id和class值，此循环执行7次
        print(row[‘id’],row[‘class’])

打印结果为：
0 5
1 4
2 13
…
6 15

写入内容

现假设我们要创建一个格式如上图所示的csv文件，则实现代码应该如下：

with open(‘name.csv’,’w’) as csvfile:
    writer = csv.DictWriter(csvfile,fieldnames=[‘id’,’class’])
    #写入列标题，即DictWriter构造方法的fieldnames参数
    writer.writeheader()
    for data in datas:
        writer.writerow({‘id’:data[0],’class’:data[1]})