Python：CSV读写（以字典的方式处理）

#!/usr/bin/env python
# coding:UTF-8


"""
@version: python3.x
@author:曹新健
@contact: [email protected]
@software: PyCharm
@file: CSV读写2.py
@time: 2018/11/13 14:27
"""
import csv

#csv的DictReader()和DictWriter()方法将以字典的方式处理csv文件，实质上是以OrderedDict方式进行处理
#默认将CSV文件的第一列作为字段名
def csvDictReader1(path):
    with open(path) as rf:
        reader = csv.DictReader(rf)
        items = list(reader)
    return items

#指定字段名，第一列也为数据
def csvDictReader2(path,fieldnames):
    with open(path) as wf:
        reader = csv.DictReader(wf,fieldnames=fieldnames)
        items = list(reader)
    return items

def csvDictWriter(path,data,fieldnames):
    with open(path,"w",newline="") as wf:
        writer = csv.DictWriter(wf,fieldnames=fieldnames)
        writer.writeheader()
        writer.writerows(data)

if __name__ == "__main__":
    fieldnames = ["aaa", "bbb", "ccc", "ddd"]
    csvItems1 = csvDictReader1("111.csv")
    csvItems2 = csvDictReader2("222.csv", fieldnames)
    csvDictWriter("333.csv",csvItems1,fieldnames)
    csvDictWriter("444.csv",csvItems2,fieldnames)