如何遍历数据目录并动态导入各子目录模块中的unpack_slices函数？

问题根源 #

你的代码无法找到image_specific_codes包，核心问题在于：

• 切换目录后，当前image子目录并未被添加到Python的模块搜索路径（sys.path）中，Python无法识别该目录下的包结构。
• importlib.import_module的package参数要求目标包处于已被Python识别的路径下，而你的场景中每个image子目录都是独立的孤立目录，默认不会被纳入搜索范围。

可行解决方案 #

方法1：用importlib.util直接加载模块文件（推荐） #

这种方法无需修改sys.path，直接通过文件路径加载模块，适配每个image子目录独立的场景：

import os
import importlib.util

# 定位根目录下的data文件夹绝对路径
data_dir = os.path.join(os.path.dirname(__file__), 'data')

for image_folder in os.listdir(data_dir):
    image_path = os.path.join(data_dir, image_folder)
    if not os.path.isdir(image_path):
        continue  # 跳过非目录文件
    
    # 拼接open_slices.py的绝对路径
    module_file_path = os.path.join(image_path, 'image_specific_codes', 'open_slices.py')
    if not os.path.exists(module_file_path):
        print(f"跳过{image_folder}：未找到open_slices.py")
        continue
    
    # 加载模块
    spec = importlib.util.spec_from_file_location(f"{image_folder}_open_slices", module_file_path)
    open_slices_module = importlib.util.module_from_spec(spec)
    spec.loader.exec_module(open_slices_module)
    
    # 获取并执行解包函数
    unpack_fn = getattr(open_slices_module, 'unpack_slices')
    unpacked_image = unpack_fn()
    
    # 后续统一分析流程
    # ...

方法2：临时修改sys.path #

将当前image子目录临时加入Python的模块搜索路径，让Python能找到image_specific_codes包：

import os
import sys
import importlib

data_dir = os.path.join(os.path.dirname(__file__), 'data')

for image_folder in os.listdir(data_dir):
    image_path = os.path.join(data_dir, image_folder)
    if not os.path.isdir(image_path):
        continue
    
    # 临时添加当前image目录到搜索路径
    sys.path.append(image_path)
    
    try:
        # 导入目标模块
        module = importlib.import_module('image_specific_codes.open_slices')
        unpack_fn = getattr(module, 'unpack_slices')
        unpacked_image = unpack_fn()
        
        # 后续统一分析流程
        # ...
    finally:
        # 移除临时路径，避免干扰后续导入操作
        sys.path.pop()

方法3：通过规范包结构导入 #

给每个image子目录添加空的__init__.py，将其转为Python包，再通过绝对路径导入：

import os
import sys
import importlib

data_dir = os.path.join(os.path.dirname(__file__), 'data')

for image_folder in os.listdir(data_dir):
    image_path = os.path.join(data_dir, image_folder)
    if not os.path.isdir(image_path):
        continue
    
    # 给image子目录添加__init__.py（如果不存在）
    init_file_path = os.path.join(image_path, '__init__.py')
    if not os.path.exists(init_file_path):
        open(init_file_path, 'w').close()
    
    # 将根目录加入搜索路径
    sys.path.append(os.path.dirname(data_dir))
    
    try:
        # 以完整包路径导入模块
        module_full_name = f"data.{image_folder}.image_specific_codes.open_slices"
        module = importlib.import_module(module_full_name)
        unpack_fn = getattr(module, 'unpack_slices')
        unpacked_image = unpack_fn()
        
        # 后续统一分析流程
        # ...
    finally:
        sys.path.pop()

关键注意点 #

• 尽量避免使用os.chdir，它会改变全局工作目录，容易引发后续文件操作的路径混乱。
• 方法1无需修改路径或添加额外文件，是最适配当前场景的方案。

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