Python文件操作总结——CSV文件的读取和写入
python中的CSV文件的读取和写入操作比较简单和直观,只需要把csv文件中数据读出到list中,在list中修改,再写入即可,至于涉及到的更多高级特性,其实一般并用不到。
首先给出_demo.csv_文件,其中包含了列名和四条样本。在CSV文件中,列名和普通的样本的处理方式并没有区别,列名一般直接放在首行即可,CSV文件会默认将首行的内容当做列名。
Sentence, Label
当一个人对别人不公平、愚蠢的时候。,6
当我爬上树摘苹果。我所站的梯子的角度不够高。这意味着梯子不太稳定。,2
在恋爱的过程中,我们每次见面,尤其是很久没有见面的时候。,1
当我卷入一场交通事故。,2
1. CSV文件的读取
首先以只读的方式(‘r’)打开文件得到文件句柄,在通过调用csv.reader()
方法,将文件句柄作为参数得到迭代器file_reader,再用这个迭代器读取csv文件中的数据即可
import csv
# 先打开文件,file_handler为打开的文件的句柄
with open('demo.csv', 'r', encoding='utf-8') as file_handler:
# 返回csv迭代器file_reader,用于迭代得到样本
file_reader = csv.reader(file_handler)
# print(type(file_reader))
# 通过迭代器,遍历csv文件中的所有样本
for sample in file_reader:
print(sample)
输出结果为:
2. CSV文件的写入
首先以写的方式打开csv文件,得到文件句柄file_handler,再以文件句柄为参数调用csv.writer()
方法,返回的对象即可用于在对应的csv文件中写入新的数据。
这里有两点需要注意:
- 文件的打开方式不一样最后在文件中写入的内容也不一样:
- 如果文件是以
'w'
方式打开,则写入时原有的内容会被清空,文件中为新写入的内容;- 如果文件是以
'a+'
方式打开,则写入方式为追加,即在原有内容的基础上写入,原有的内容不会被覆盖。- 为了避免样本之间存在空行,调用open()函数时,应该指定参数
newline=''
import csv
# 以'w'方式打开
with open('demo.csv', 'w', encoding='utf-8', newline='') as file_handler:
file_writer = csv.writer(file_handler)
row = ['a single row', 'single label']
rows = [['row 1', 'label 1'],
['row 2', 'label 2'],
['row 3', 'label 3']]
file_writer.writerow(row) # 写入一行
file_writer.writerows(rows) # 写入多行
# 以'a+'方式打开
with open('demo.csv', 'a+', encoding='utf-8', newline='') as file_handler:
file_writer = csv.writer(file_handler)
row = ['a single row', 'single label']
rows = [['row 1', 'label 1'],
['row 2', 'label 2'],
['row 3', 'label 3']]
file_writer.writerow(row)
file_writer.writerows(rows)