Python 模块-csv

数据存储，操作csv

import csv

header = ["ID", "UserName", "PassWord"]
'''
    第一种方式写入csv文件
'''
# rows = [{"ID": "1", "UserName": "增加", "PassWord": "1234"}, {"ID": "1", "UserName": "增加", "PassWord": "1234"}]
# with open("data/testCSV.csv", 'w') as f:
#     f_cvs = csv.DictWriter(f, header)
#     f_cvs.writeheader()
#     f_cvs.writerows(rows)

'''
    第二种方式写入csv文件
'''
# rows = [(1, "zengjia", "123"), (2, "zengjia", "123")]
# with open("data/testCSV2.csv", 'w') as f:
#     f_cvs = csv.writer(f)
#     f_cvs.writerow(header)
#     f_cvs.writerows(rows)

'''
    保存csv文件
'''

# 读取CSV文件
# with open('data/testCSV.csv') as f:
#     f_csv = csv.reader(f)
#     headers = next(f_csv)
#     print(headers)
#     for row in f_csv:
#         print(row)

'''
    存在空行记录时，不可以转换成nametuple
'''

# from collections import namedtuple
# with open('data/testCSV.csv') as f:
#     f_csv = csv.reader(f)
#     heading = next(f_csv)
#     Row = namedtuple('Row', heading)
#     for r in f_csv:
#         row = Row(*r)
#         print(row)

'''
    DictReader 读取CSV文件
'''
# with open('data/testCSV.csv') as f:
#     f_csv = csv.DictReader(f)
#     for row in f_csv:
#         print(row.get('ID'), row.get('UserName'), row.get('PassWord'))