python在CSV中添加标题行的用法

1,027次阅读
没有评论

python在CSV中添加标题行的用法

有时候,电子表格中没有标题行,但是您确实希望所有列都有一个列标题。本例中,您可以使用脚本添加列标题。

说明如何使用脚本添加列标题,您需要修改一下输入文件:

(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免费测试

相关文章:

版权声明:wuyou2021-05-04发表,共计1070字。
新手QQ群:570568346,欢迎进群讨论 Python51学习