4_4 写入csv文件

"""写入csv文件"""


import csv

# 方法一：
def write_csv_demo1():
    headers = ['username', 'age', 'height']
    values = [
        ('张三', 18, 180),
        ('李四', 19, 190),
        ('王五', 20, 120)
    ]
    with open('csvwriter1.csv', 'w', encoding='gbk', newline='') as fp:
        fp.writer = csv.writer(fp)
        fp.writer.writerow(headers)
        fp.writer.writerows(values)

# 方法二：
def write_csv_demo2():
    headers = ['username', 'age', 'height']
    values = [
        {'username': '张三', 'age': 18, 'height': 180},
        {'username': '李四', 'age': 19, 'height': 190},
        {'username': '王五', 'age': 20, 'height': 120}
    ]
    with open('csvwriter2.csv', 'w', encoding='utf-8', newline='') as fp:
        write = csv.DictWriter(fp, headers)
        write.writeheader() # 写入表头信息
        write.writerows(values)

if __name__ == "__main__":
    write_csv_demo1()
    write_csv_demo2()