文章 > Python基础教程 > python如何在CSV中添加标题行

python如何在CSV中添加标题行

头像

小妮浅浅

2021-03-01 14:53:30574浏览 · 0收藏 · 0评论

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

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

(1)在电子表格程序中打开 supplier_data.csv。

(2)删除文件中的第一行(即包含列标题的标题行)。

(3)将文件保存为 supplier_data_no_header_row.csv。

2、要使用基础 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 行代码将所有数据行写入输出文件,放在标题行下面。

3、要运行这个脚本,在命令行中输入以下命令,然后按回车键:

python 12csv_reader_add_header_row.py supplier_data_no_header_row.csv\
output_files\12output.csv

可以打开输出文件 12output.csv 查看一下结果。

以上就是python在CSV中添加标题行方法,希望能对大家有所帮助。更多Python学习指路:python基础教程

关注公众号

关注公众号,随时随地在线学习

本教程部分素材来源于网络,版权问题联系站长!

底部广告图 底部广告图按钮