如何编写Python脚本调用现有脚本批量处理HDF5文件？

这里提供三种实用的实现思路，按需选择：

方案一：修改原脚本支持命令行参数，再批量调用 #

这种方式保留原脚本的独立使用能力，适合需要单独运行原脚本处理单个文件的场景。

1. 修改原脚本，让它能接收命令行传入的文件名：

import h5py
import sys

# 校验参数数量
if len(sys.argv) != 2:
    print("使用方式: python your_script.py <hdf5_file_name>")
    sys.exit(1)

# 获取传入的文件名
target_file = sys.argv[1]
readFile = h5py.File(target_file, 'r')

# 原脚本中的数学运算、输出F的代码...

readFile.close()

2. 编写批量调用脚本，循环生成文件名并执行原脚本：

import subprocess

# 遍历01到10的文件序号
for idx in range(1, 11):
    # 生成两位格式的序号（如01、02...10）
    file_suffix = f"{idx:02d}"
    file_name = f"FileName_{file_suffix}"
    # 调用原脚本
    subprocess.run(["python", "your_script.py", file_name], check=True)

方案二：直接在原脚本中添加循环逻辑 #

如果不需要单独运行原脚本处理单个文件，直接在原脚本里加循环是最高效的方式，避免多次启动Python进程的开销：

import h5py

# 循环处理FileName_01到FileName_10
for idx in range(1, 11):
    file_suffix = f"{idx:02d}"
    file_name = f"FileName_{file_suffix}"
    
    with h5py.File(file_name, 'r') as readFile:  # 使用with语句自动关闭文件，更安全
        # 原脚本中的数学运算、输出F的代码...

方案三：用Shell脚本批量调用（适合Linux/macOS/WSL环境） #

如果熟悉Shell命令，不用修改Python代码，直接写个Shell脚本循环调用即可：

#!/bin/bash

# 循环处理01到10的文件
for num in {01..10}
do
    python your_script.py "FileName_$num"
done

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