"""csv两种写入操作"""
import csv
def write_csv_demo():
headers = ["username", "age", "height"]
values = {
("张三", 18, 180),
("李四", 19, 190),
("王五", 20, 160)
}
# 写入编码utf-8防止乱码,newline默认\n换行,防止换行让它等于空字符串
with open("classroom.csv", "w", encoding="utf-8", newline="") as fp:
# 创建一个文件句柄对象
writer = csv.writer(fp)
# 写入一行writerow
writer.writerow(headers)
# 写入多行writerows
writer.writerows(values)
def write_csv_site():
headers = ["username", "age", "height"]
values = [
{"username": "张三", "age": 18, "height": 180},
{"username": "李四", "age": 19, "height": 190},
{"username": "王五", "age": 20, "height": 160},
]
with open("classroom1.csv", "w", encoding="utf-8", newline="") as fp:
# DictWriter需要传入一个文件句柄和字典的key列表
writer = csv.DictWriter(fp, headers)
# 正式写入需要调用writeheader方法
writer.writeheader()
writer.writerows(values)
if __name__ == '__main__':
write_csv_site()