Python操作CSV文件详解:读取与写入实战
本文详细介绍Python中csv模块的使用方法,涵盖如何读取CSV文件的全部数据及指定列。通过对比reader和DictReader两种读取方式,解析字符串类型处理及按列名查询的技巧,并演示如何追加写入新数据到CSV文件,帮助开发者高效处理表格数据。
CSV(Comma-Separated Values,逗号分隔值)是一种以纯文本形式存储表格数据的格式。例如,如下所示的表格结构:

该表格可以保存为CSV文件,其实际文件内容如下:
No.,Name,Age,Score 1,mayi,18,99 2,jack,21,89 3,tom,25,95 4,rain,19,80
假设上述CSV文件被保存为“test.csv”。
1. 读取文件
如何使用Python像操作Excel一样提取某一列(即一个字段)的数据?利用Python内置的csv模块,主要有两种实现方法:
第一种方法是使用reader函数。该函数接收一个可迭代对象(如csv文件对象),并返回一个生成器,从而可以解析出CSV的内容。例如,以下代码用于读取CSV的全部数据,按行进行输出:
#!/usr/bin/python3
# -*- conding:utf-8 -*-
__author__ = 'mayi'
import csv
#读
with open("test.csv", "r", encoding = "utf-8") as f:
reader = csv.reader(f)
rows = [row for row in reader]
print(rows)
运行结果为:
[['No.', 'Name', 'Age', 'Score'], ['1', 'mayi', '18', '99'], ['2', 'jack', '21', '89'], ['3', 'tom', '25', '95'], ['4', 'rain', '19', '80']]
若要提取其中的某一列,可以使用以下代码:
#!/usr/bin/python3
# -*- conding:utf-8 -*-
__author__ = 'mayi'
import csv
#读取第二列的内容
with open("test.csv", "r", encoding = "utf-8") as f:
reader = csv.reader(f)
column = [row[1] for row in reader]
print(column)
运行结果为:
['Name', 'mayi', 'jack', 'tom', 'rain']
需要注意的是,从CSV读取的数据均为str类型。这种方法要求事先知道列的序号(例如Name位于第2列),无法直接通过标题“Name”进行查询。此时,可以采用第二种方法:
第二种方法是使用DictReader。与reader函数类似,它接收一个可迭代对象并返回生成器,但返回的每个单元格都封装在一个字典的值中,而字典的键则是该单元格的标题(即列名)。通过以下代码可以查看DictReader的结构:
# -*- conding:utf-8 -*-
__author__ = 'mayi'
import csv
#读
with open("test.csv", "r", encoding = "utf-8") as f:
reader = csv.DictReader(f)
column = [row for row in reader]
print(column)
运行结果为:
[{'No.': '1', 'Age': '18', 'Score': '99', 'Name': 'mayi'},
{'No.': '2', 'Age': '21', 'Score': '89', 'Name': 'jack'},
{'No.': '3', 'Age': '25', 'Score': '95', 'Name': 'tom'},
{'No.': '4', 'Age': '19', 'Score': '80', 'Name': 'rain'}]
若希望使用DictReader读取CSV的特定列,可以通过列标题进行查询:
#!/usr/bin/python3
# -*- conding:utf-8 -*-
__author__ = 'mayi'
import csv
#读取Name列的内容
with open("test.csv", "r", encoding = "utf-8") as f:
reader = csv.DictReader(f)
column = [row['Name'] for row in reader]
print(column)
运行结果为:
['mayi', 'jack', 'tom', 'rain']
2. 写入文件
在读取文件时,我们将CSV数据读入列表;而在写入文件时,则是将列表中的元素写入到CSV文件中。
#!/usr/bin/python3
# -*- conding:utf-8 -*-
__author__ = 'mayi'
import csv
#写:追加
row = ['5', 'hanmeimei', '23', '81']
out = open("test.csv", "a", newline = "")
csv_writer = csv.writer(out, dialect = "excel")
csv_writer.writerow(row)
运行结果为:

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持本站。
本站声明
本文内容仅供参考,不代表本站观点。转载或引用请注明出处。如发现内容问题,请通过联系我们页面反馈。













