参考:https://www.jianshu.com/p/f59f101654cf
import csv
class CSVFileReadOrWrite:
@classmethod
# 读取csv文件返回一个list-list格式
def read_csv_file(cls, file_path):
rows = []
# with进入执行上下文,退出的时候会自动关闭文件
with open(file_path, 'r') as csv_file:
# 读取csv文件中每行的列表,将每行读取的值作为列表返回
csv_read = csv.reader(csv_file)
for row in csv_read:
rows.append(row)
return rows
@classmethod
# 给csv文件写入一行数据【默认打开模式选择a模式,表示如果文件存在则追加】
def write_csv_file(cls, file_path, row_list, mode="a"):
# 以newline=''打开csv文件进行追加,newline可避免空行
with open(file_path, mode, newline='') as csv_file:
# 设定写入模式
csv_write = csv.writer(csv_file)
# 写入具体内容
csv_write.writerow(row_list)
官方提供了一个读写CSV文件的库