Python批量处理目录下XYZ文件生成同名TXT文件需求

核心需求回顾 #

• 遍历指定目录下所有.xyz格式文件
• 对每个文件执行：
  • 跳过第二行内容
  • 在文件开头添加两行固定文本：MOLECULE和1
  • 将所有行中的字母H替换为1
  • 输出同名.txt文件到指定的目标目录

完整实现代码 #

import os

# 配置路径，替换成你实际的源目录和目标目录
SOURCE_DIR = "C:/你的源文件目录路径"
TARGET_DIR = "C:/你的目标文件目录路径"

# 确保目标目录存在，不存在则自动创建
os.makedirs(TARGET_DIR, exist_ok=True)

# 遍历源目录下所有.xyz文件
for filename in os.listdir(SOURCE_DIR):
    if filename.endswith(".xyz"):
        # 拼接源文件和目标文件的完整路径
        source_path = os.path.join(SOURCE_DIR, filename)
        target_filename = os.path.splitext(filename)[0] + ".txt"
        target_path = os.path.join(TARGET_DIR, target_filename)
        
        # 处理单个文件，用with语句自动管理文件开闭
        with open(source_path, 'r', encoding='utf-8') as src_file, open(target_path, 'w', encoding='utf-8') as target_file:
            # 写入开头固定文本
            target_file.write("MOLECULE\n")
            target_file.write("1\n")
            
            # 读取第一行，替换H为1后写入
            first_line = src_file.readline()
            target_file.write(first_line.replace('H', '1'))
            
            # 跳过第二行
            src_file.readline()
            
            # 读取剩余所有行，逐行替换H为1后写入
            for line in src_file:
                modified_line = line.replace('H', '1')
                target_file.write(modified_line)

关键代码解释 #

1. 路径与目录处理：

   • 直接修改SOURCE_DIR和TARGET_DIR为你实际的文件夹路径即可
   • os.makedirs(...)会自动创建目标目录，避免因目录不存在导致的报错

2. 批量文件遍历：

   • os.listdir(SOURCE_DIR)列出源目录所有文件，通过endswith(".xyz")精准筛选需要处理的文件

3. 安全高效的文件操作：

   • 使用with语句自动处理文件的打开和关闭，避免手动操作可能出现的文件泄漏问题
   • 同时打开源文件和目标文件，边读边写，不需要中间临时文件，处理7000个文件时效率更高

4. 内容处理逻辑：

   • 先写入指定的开头文本
   • 读取第一行并完成替换后写入
   • 调用一次readline()直接跳过第二行
   • 遍历剩余所有行，逐行替换H为1后写入目标文件

对你原有代码的优化 #

• 去掉了反复打开关闭文件的冗余操作，一次完成所有处理步骤
• 移除了中间临时文件，减少磁盘IO开销
• 用批量遍历逻辑替代单文件处理，直接覆盖所有7000个文件的需求

内容的提问来源于stack exchange，提问作者Mat