套用【Python编程快速上手—让繁琐工作自动化】一书中捋清思路的方法,从思路到代码行为、从抽象(想法)到具体(方法)、递进列出代码需要实现的内容。
总的来说,该程序必须做到以下几点:
- 找出当前工作目录中的所有 CSV 文件
- 读取每个文件的全部内容
- 修改第一行,将内容写入一个新的 CSV 文件
在代码层面上,这意味着该程序需要做到以下几点:
- 循环遍历从 os.listdir()得到的文件列表,跳过非 CSV 文件
- 创建一个 CSV Reader 对象,读取并修改该文件的内容
- 创建一个 CSV Writer 对象,将修改好的数据写入新文件
#coding=UTF-8
import os
import csv
root = r'D:\03code\python\csv_data'#需要处理csv文件的根目录
root_root, root_dir_name = os.path.