python文件处理之CSV

'''
csv的处理
'''

# 读取csv文件
import csv
with open("config/person_info.csv", encoding="unicode_escape") as res_file:
    # csv.reader返回的是一个迭代类型，索引应该用循环来打印信息才行
    csv_file = csv.reader(res_file)
    # 通过for循环一次取出每一行数据
    print(csv_file)
    for row in csv_file:
        # 依次打印每一行的数据，每一行的数据类型存放在列表中
        print("row>>>",row)
        
# 以列表的方式写入csv文件
import csv
header = ['username', 'eamil', 'tel']
data = [
    ['商店客户端', 'lisa001', '123456'],
    ['开发者后台', 'lisa002', '123456'],
    ['管理员后台', 'lisa003', '123456']
]
# 以写（w）的方式，编码格式为utf-8 ,行位不添加换行符的方式，打开目标文件person_info_001.csv
with open("config/person_info_001.csv", mode='w', encoding="UTF-8", newline="") as res_file:
    # 创建csv_file 写入操作的控制器
    csv_file = csv.writer(res_file)
    # 通过csv写入控制器，写入一行数据（表头）
    csv_file.writerow(header)
    # 通过csv写入控制器，写入多行数据
    csv_file.writerows(data)

# 以字典的方式写入
import csv
# 定义一个列表header存放表头
header = ['username', 'email', 'tel']
# 定义一个字典data存放数据
data = [
    {"username": "商店客户端", "email": "lisa001@qq.com", "tel": "123456"},
    {"username": "开发者后台", "email": "lisa002@qq.com", "tel": "123456"},
    {"username": "管理员后台", "email": "lisa003@qq.com", "tel": "123456"}
]
with open("config/person_info_002.csv",mode="w", encoding="UTF-8", newline="") as res_file:
    # 创建csv_file 写入操作的控制器
    csv_file = csv.DictWriter(res_file, header)
    # 通过csv写入控制器，写入一行数据（表头）
    csv_file.writeheader()
    # 通过csv写入控制器，写入多行数据
    csv_file.writerows(data)