CSV是数据处理中经常会用到的一种文件格式,一般也被叫做逗号分隔值文件,是一种纯文本文件,使用特定的结构来排列表格数据,可以使用 Excel 软件或者文本文档打开 。
csv模块 是 python标准库模块,无需手动安装,可以直接导入 import csv # 导入csv模块
,此篇文章介绍csv模块的基本使用方法。
file = open("xxx.csv", "w", newline="") # 打开文件
writer = csv.writer(file) # 初始化写入对象
writer.writerow()
& writer.writerows() # 写入数据
csv 文件写入用到的模块方法是 csv.writer(),该方法原型:csv.writer(csvfile, dialect='excel', **fmtparams)
writerow([]) 单行写入,一次只能写入一行数据
示例:
with open("test.csv", "w") as f: # 打开文件,后缀为.csv
writer = csv.writer(f) # 初始化写入对象
writer.writerow(["1", "1000"]) # 写入数据
writerows([(),(),()]) 多行写入,一次写入多行数据
示例:
lst = [("1", "1000"), ("2", "998"), ("3", "1005")]
with open("test.csv", "w", newline="") as f: # 打开文件,后缀为.csv
writer = csv.writer(f) # 初始化写入对象
writer.writerows(lst) # 写入数据
注意: 需要加入newline=“”,否则每写入一行会有一空行;windows系统会出现bug,所以需要添加该语句,linux系统不存在该bug。
DictWriter(file,[])两个参数为 打开的文件对象和csv 的列头行列表
示例:
data_list = [
{"序号": "0", "数据": 1000},
{"序号": "1", "数据": 998},
{"序号": "2", "数据": 1005},
] # 定义数据列表,列表的每个元素都是字典
header_list = ["序号", "数据"] # 创建列头行header 列表
with open("test.csv", "w", newline="") as f: # 打开文件,后缀为.csv
writer = csv.DictWriter(f,header_list) # 初始化写入对象,将 header 列表作为参数传入
writer.writeheader() # 写入列头
writer.writerows(data_list) # 写入数据
注意:
在写入正式数据前,需使用 writeheader()方法 写入列头
file = open("xxx.csv", "r", newline="") # 打开文件
reader = csv.reader(file)
list(reader)
将reader转化为list,方便进行遍历reader(),以列表的形式返回读取的数据
示例:
with open('test.csv', 'r', newline='') as f: # 打开文件
reader = csv.reader(f) # 创建reader对象
for r in reader: # 输出信息
print(r)
['1', '1000']
['2', '998']
['3', '1005']
DictReader(),以字典的形式返回读取的数据
示例:
with open('test.csv', 'r', newline='') as f: # 打开文件
reader = csv.DictReader(f) # 创建DictReader对象
for r in reader: # 输出信息
print(r)
{'序号': '0', '数据': '1000'}
{'序号': '1', '数据': '998'}
{'序号': '2', '数据': '1005'}
注意: 如果直接打印reader,会返回
,需要对其进行转化 list(reader)