python csv的基本操作

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)

附

https://docs.python.org/2/library/csv.html#module-csv.

参考

https://blog.csdn.net/u012162613/article/details/41915859.
https://blog.csdn.net/u012162613/article/details/41915859.