有时候,电子表格中没有标题行,但是您确实希望所有列都有一个列标题。本例中,您可以使用脚本添加列标题。
说明如何使用脚本添加列标题,您需要修改一下输入文件:
(1)在电子表格程序中打开 supplier_data.csv。
(2)删除文件中的第一行(即包含列标题的标题行)。
(3)将文件保存为 supplier_data_no_header_row.csv。
使用基础 Python 添加列标题,在文本编辑器中输入下列代码,然后将文件保存为 12csv_reader_add_header_row.py:
#!/usr/bin/env python3
import csv
import sys
input_file = sys.argv[1]
output_file = sys.argv[2]
with open(input_file, 'r', newline='') as csv_in_file:
with open(output_file, 'w', newline='') as csv_out_file:
filereader = csv.reader(csv_in_file)
filewriter = csv.writer(csv_out_file)
header_list = ['Supplier Name', 'Invoice Number',\
'Part Number', 'Cost', 'Purchase Date']
filewriter.writerow(header_list)
for row in filereader:
filewriter.writerow(row)
第 10 行代码创建了列表变量 header_list,其中包含了要作为列标题的 5 个字符串。第 12 行代码将这些列表值写入输出文件的第一行。同样,第 14 行代码将所有数据行写入输出文件,放在标题行下面。
运行这个脚本,在命令行中输入以下命令,然后按回车键:
python 12csv_reader_add_header_row.py supplier_data_no_header_row.csv\
output_files\12output.csv
可以打开输出文件 12output.csv 查看一下结果。
以上就是python在CSV中添加标题行方法,希望能对大家有所帮助。
神龙|纯净稳定代理IP免费测试>>>>>>>>天启|企业级代理IP免费测试>>>>>>>>IPIPGO|全球住宅代理IP免费测试