Python数据分析基础——CSV文件——添加标题行

参考文献：《Python数据分析基础》

前言

有时，电子表格中没有标题行，但确实希望所有列都有列标题。在这种情况下，可以使用脚本添加列标题。
例如，在supplier_data_no_header_row.csv文件中，我们希望添加列标题使文件格式完整。

创建脚本

在文本编辑器中输入一段代码，然后将文件保存为：12csv_reader_add_header_row.csv

#!/usr/bin/env python3

import sys
import csv

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', 'Invoice Number', 'Part Number', 'Cost', 'Purchare Date']
        filewriter.writerow(header_list)
        for row in filereader:
            filewriter.writerow(row)

脚本代码注释

header_list = ['Supplier', 'Invoice Number', 'Part Number', 'Cost', 'Purchare Date']
filewriter.writerow(header_list)

这两行代码是实现添加标题行的关键。第一行代码创建了header_list列表变量，其中包含了作为列标题的5个字符串。第二行代码将这些列表值写入输出文件的第一行。

filewriter.writerow(header_list)
        for row in filereader:
            filewriter.writerow(row)

这部分代码是将所有数据写入输出文件，放在标题行下面。

运行脚本

在命令行输入以下命令，然后按回车键：

查看结果

（左为添加前，右为添加后）

结语

本文介绍了添加标题行的相关知识及操作，总结了书上内容和实际操作。