1 读取数据
csv.reader
csv.reader传入的可以是列表或者文件对象,返回的是一个可迭代的对象,需要使用for循环遍历。
path = "C:\\Users\\A539\\Desktop\\1.csv"
with open(path, 'r') as fp:
lines = csv.reader(fp)
for line in lines:
print(line)
print(type(line))
line的格式为list
2 写入数据
csv.writer
将一个列表写入csv文件
list1 = [100, 200, 300, 400, 500]
list2 = [[500, 600, 700, 800, 900],
[50, 60, 70, 80, 90]]
with open(path, 'w',newline='')as fp:
writer = csv.writer(fp)
# 写入一行
writer.writerow(list1)
# 写入多行
writer.writerows(list2)
不加newline = ''会导致每行之间有一行空行
csv.DictWriter
写入字典
head = ['aa', 'bb', 'cc', 'dd', 'ee']
lines = [
{'aa': 10 , 'bb': 20, 'cc': 30, 'dd': 40, 'ee': 50},
{'aa': 100, 'bb': 200, 'cc': 300, 'dd': 400, 'ee': 500},
{'aa': 1000, 'bb': 2000, 'cc': 3000, 'dd': 4000, 'ee': 5000},
{'aa': 10000, 'bb': 20000, 'cc': 30000, 'dd': 40000, 'ee': 50000},
]
with open(path, 'w',newline='')as fp:
dictwriter = csv.DictWriter(fp, head)
dictwriter.writeheader()
with open(path, 'w', newline='')as fp:
dictwriter = csv.DictWriter(fp, head)
dictwriter.writeheader()
dictwriter.writerows(lines)
不覆盖原有内容写入
上述的写入都会覆盖原有的内容,要想保存之前的内容,将新内容附加到后面,只需要更改标志为’a+’
with open(path, 'a+', newline='')as fp:
dictwriter = csv.DictWriter(fp, head)
dictwriter.writeheader()
dictwriter.writerows(lines)
最后多说一句,我是一名python开发工程师,这里有我自己整理了一套最新的系统学习教程,包括从基础python脚本到web开发、爬虫、数据分析、数据可视化、机器学习等。想要这些资料的可以关注小编,并在后台私信小编:“01”即可领取。