python代码:A文件夹下和B文件夹有有同样的txt文件,每个txt文件都只有一行数字。将A文件夹下的txt文件第一行数字加到B文件夹下同前缀的的txt文件下的第一行
时间: 2024-09-08 09:04:39 浏览: 33
你可以使用Python的os和fileinput模块来实现这个操作。这里是一个简单的脚本示例:
```python
import os
from fileinput import input
# 定义两个文件夹路径
folder_a = 'A'
folder_b = 'B'
# 遍历A文件夹中的所有.txt文件
for filename in os.listdir(folder_a):
if filename.endswith('.txt'):
with open(os.path.join(folder_a, filename), 'r') as a_file:
# 读取A文件的第一行数字
line_a = int(a_file.readline())
# 获取B文件夹对应文件名(保留相同的前缀)
prefix = filename[:-4] # 去除.txt
b_filename = prefix + '.txt'
b_path = os.path.join(folder_b, b_filename)
# 检查B文件是否存在并打开进行写入
try:
with open(b_path, 'a') as b_file:
# 将A文件的第一行数字追加到B文件的开头
b_file.write(str(line_a) + '\n')
except FileNotFoundError:
print(f"{b_filename} not found in {folder_b}")
阅读全文