Python编程练习5:在文件头部添加新行

工作的时候会遇到大量txt文件，有时需要批量在文件头部添加信息，这里用python写个小程序来完成这个工作

##python批量添加文件头部
import os

def add_title(filename): # 添加首行title
    with open(filename, 'r+', encoding = 'utf-8') as f:
        content = f.read()
        f.seek(0,0)
        f.write('title\n'+content)

for dirpath, dirnames, filenames in os.walk('.'):
    for filename in filenames:
        name = os.path.splitext(filename)
        if name[1] == '.txt':
            add_title(filename)

注意：
os.walk()可以遍历文件夹中的所有子文件夹及子文件, walk函数返回一个元组(dirpath, dirnames, filenames)

dirpath是文件夹路径
dirnames是文件夹名称
filenames是文件名称

这个程序可以一次性处理完文件夹下所有的txt文件，使用起来极其方便

Snipaste_2022-09-22_16-44-00.png