Python数据分析基础 |
您所在的位置:网站首页 › pycharm如何生成csv文件 › Python数据分析基础 |
参考文献:《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)这部分代码是将所有数据写入输出文件,放在标题行下面。 运行脚本在命令行输入以下命令,然后按回车键: (左为添加前,右为添加后) 本文介绍了添加标题行的相关知识及操作,总结了书上内容和实际操作。 |
今日新闻 |
推荐新闻 |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |