python文件操作——CSV文件的读取和写入

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文件中写入新的数据。

这里有两点需要注意：

1. 文件的打开方式不一样最后在文件中写入的内容也不一样：
   1. 如果文件是以'w'方式打开，则写入时原有的内容会被清空，文件中为新写入的内容；
   2. 如果文件是以'a+'方式打开，则写入方式为追加，即在原有内容的基础上写入，原有的内容不会被覆盖。
2. 为了避免样本之间存在空行，调用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)