Python数据分析基础

您所在的位置:网站首页 pycharm如何生成csv文件 Python数据分析基础

Python数据分析基础

2024-07-14 15:02| 来源: 网络整理| 查看: 265

参考文献:《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