写一段python程序:路径G:\drought\data\data1下有文件夹”station“和”stationhh“。把”station“下”52765.txt“的第一行数据复制到”stationhh“下所有txt文件的第一行之前。
时间: 2024-09-11 21:11:14 浏览: 43
forecastVeg:一种用Python预测遥感植被健康的机器学习方法
以下是一个Python脚本示例,用于完成您的要求:
```python
import os
# 定义源目录和目标目录
source_dir = r"G:\drought\data\data1\station"
target_dir = r"G:\drought\data\data1\stationhh"
# 检查源目录和目标目录是否存在
if not os.path.exists(source_dir) or not os.path.exists(target_dir):
print("指定的路径不存在,请检查后再试。")
else:
# 源文件路径
source_file = os.path.join(source_dir, "52765.txt")
# 检查源文件是否存在
if not os.path.isfile(source_file):
print("源文件不存在,请检查后再试。")
else:
# 读取源文件的第一行
with open(source_file, 'r') as file:
first_line = file.readline().strip() # 读取第一行并去除尾部的换行符
# 遍历目标目录下的所有txt文件,并将第一行数据写入每个文件之前
for filename in os.listdir(target_dir):
if filename.endswith(".txt"):
target_file = os.path.join(target_dir, filename)
with open(target_file, 'r+') as file: # 打开文件进行读写
content = file.readlines() # 读取所有行
file.seek(0) # 移动文件指针到文件开头
file.write(first_line + '\n' + ''.join(content)) # 写入源文件的第一行和原文件内容
file.truncate() # 截断文件,只保留新写入的内容
print("操作完成。")
```
这段代码首先定义了源目录和目标目录的路径,并检查这些路径是否存在。如果都存在,它会读取`station`文件夹下`52765.txt`文件的第一行,并将这一行数据写入`stationhh`文件夹下所有txt文件的第一行之前。注意,这段代码会覆盖目标文件原有的内容,请确保这是您想要的操作。
阅读全文