当前位置:嗨网首页>书籍在线阅读

08-项目_从CSV文件中删除标题行

  
选择背景色: 黄橙 洋红 淡粉 水蓝 草绿 白色 选择字体: 宋体 黑体 微软雅黑 楷体 选择字体大小: 恢复默认

16.2 项目:从CSV文件中删除标题行

假设你有一个繁琐的任务,要删除几百个CSV文件的第一行。也许你会将它们送入一个自动化的过程,该过程只需要数据,不需要每列顶部的标题行。可以在Excel中打开每个文件,删除第一行,并重新保存该文件,但这需要几小时。让我们写一个程序来做这件事。

该程序需要打开当前工作目录中所有扩展名为.csv的文件,读取CSV文件的内容,并除掉第一行的内容以重新写入同名的文件。这将用新的、无标题行的内容替换CSV文件的旧内容。

警告: 与往常一样,当你写程序修改文件时,一定要先备份这些文件,以防你的程序没有按期望的方式工作。你不希望意外删除原始文件。

总的来说,该程序需要完成以下任务。

1.找出当前工作目录中的所有CSV文件。

2.读取每个文件的全部内容。

3.跳过第一行,将内容写入一个新的CSV文件。

在代码层面上,这意味着需要执行以下操作。

1.循环遍历从 os.listdir() 得到的文件列表,跳过非CSV文件。

2.创建一个CSV reader 对象来读取该文件的内容,并利用 line_num 属性确定要跳过哪一行。

3.创建一个CSV writer 对象,将读入的数据写入新文件。

针对这个项目,打开一个新的文件编辑器窗口,并将其保存为removeCsvHeader.py。