用python给所有txt文件加入一列为文件名

Question

用python给所有txt文件加入一列为文件名
数据如下：
文件为D00002.txt
D00002 #第一行为文件名
D00002 1.000000
D00005 0.628264
D02011 0.628264
D08646 0.554702
D08647 0.554702
D03685 0.537267

更改为
D00002 D00002 1.000000
D00002 D00005 0.628264
D00002 D02011 0.628264
D00002 D08646 0.554702
D00002 D08647 0.554702
D00002 D03685 0.537267
把第一行删掉
求帮忙，谢谢！感谢！

Answer 1

import os

# 指定文件夹路径
dir_path = '/path/to/your/files/'

# 获取该文件夹下所有的.txt文件
txt_files = [f for f in os.listdir(dir_path) if f.endswith('.txt')]

# 遍历所有的.txt文件
for file_name in txt_files:
    # 读取原始文件内容
    with open(os.path.join(dir_path, file_name), 'r') as f:
        lines = f.readlines()[1:]  # 删除第一行

    # 生成新的文件内容
    new_lines = []
    for line in lines:
        new_lines.append(file_name.replace('.txt', '') + ' ' + line)  # 在每行前面添加文件名

    # 写回新的文件内容
    with open(os.path.join(dir_path, file_name), 'w') as f:
        f.writelines(new_lines)