Python 写入 CSV 文件,去除表头元素
可以使用 csv.writer 或 csv.DictWriter 来写入 CSV 文件。示例代码如下:
使用 csv.writer:
import csv
data = [['Jack', '23', 'Male'],
['Lucy', '25', 'Female'],
['Tom', '20', 'Male']]
with open('data.csv', 'w', newline='') as file:
writer = csv.writer(file)
for row in data:
writer.writerow(row[1:]) # 写入除表头的元素
使用 csv.DictWriter:
import csv
data = [{'name': 'Jack', 'age': '23', 'gender': 'Male'},
{'name': 'Lucy', 'age': '25', 'gender': 'Female'},
{'name': 'Tom', 'age': '20', 'gender': 'Male'}]
with open('data.csv', 'w', newline='') as file:
fieldnames = data[0].keys() # 表头
writer = csv.DictWriter(file, fieldnames=fieldnames)
writer.writeheader() # 写入表头
for row in data:
del row['name'] # 删除name列
writer.writerow(row) # 写入除name列的元素
原文地址: https://www.cveoy.top/t/topic/otRJ 著作权归作者所有。请勿转载和采集!