csv 文件,逗号分割文件。
0. 读取 csv 到 list
from csv import reader
def load_csv(csvfile):
dataset = []
with open(csvfile, 'r') as f:
csv_reader = reader(f)
for row in csv_reader:
if not row:
dataset.append(row)
return dataset
1. 写入并生成 csv 文件
注意这里是写入并生成,而非创建并写入,也即可自动创建一个不存在的 csv 文件。
import csv
with open('test.csv', 'w') as f:
writer = csv.writer(f)
# 写入表头,表头是单行数据
writer.writerow(['name', 'age', 'tel'])
data = [
('zhangsan', 20, 'xxxx'),
('lisi', 22, 'xxxx')
]
# 写入这些多行数据
writer.writerows(data)
注意文件的打开模式,如果with open('', 'wb')
的方式打开,向其中写入字符内容时,很容易出现TypeError: a bytes-like object is required, not 'str' when writing to a file
的类型错误。详细内容见python 3.5: TypeError: a bytes-like object is required, not ‘str’ when writing to a file