如何通过CLI参数在Python中导入指定模块及类？

当然可以实现！Python标准库自带的importlib工具就能完美搞定这个需求——通过命令行参数动态导入模块和类，完全适配你机器学习模型调用的场景。下面给你一步步的具体实现方案：

1. 准备你的模型模块 #

首先确保你的模型模块（比如model_1.py）和主脚本在同一目录下，或者处于Python能搜索到的路径中。模块里要包含你要调用的类，比如：

# model_1.py
class ClassName:
    def __init__(self):
        # 模型初始化逻辑
        pass
    
    def predict(self, input_data):
        # 模型预测逻辑
        return f"model_1 处理数据：{input_data}"

2. 编写主脚本（script.py） #

用importlib.import_module动态导入指定模块，再通过getattr获取目标类，代码示例如下：

import sys
import importlib

def main():
    # 检查命令行参数是否合法
    if len(sys.argv) != 2:
        print("使用方式：python3 script.py <模型模块名>")
        sys.exit(1)
    
    # 获取命令行传入的模块名
    target_module = sys.argv[1]
    
    try:
        # 动态导入模块
        model_module = importlib.import_module(target_module)
        # 从模块中获取目标类（这里假设类名固定为ClassName）
        ModelClass = getattr(model_module, "ClassName")
        
        # 实例化类并使用
        model = ModelClass()
        result = model.predict("测试输入")
        print(result)
        
    except ModuleNotFoundError:
        print(f"错误：找不到名为 {target_module} 的模块")
        sys.exit(1)
    except AttributeError:
        print(f"错误：模块 {target_module} 中不存在 ClassName 类")
        sys.exit(1)

if __name__ == "__main__":
    main()

3. 运行脚本 #

在命令行执行以下命令即可调用指定模型：

python3 script.py model_1

进阶扩展：动态指定类名 #

如果你的不同模型模块里的类名不一样，也可以把类名作为第二个命令行参数传入，修改后的主脚本关键部分如下：

# 修改参数检查逻辑
if len(sys.argv) != 3:
    print("使用方式：python3 script.py <模型模块名> <类名>")
    sys.exit(1)

target_module = sys.argv[1]
target_class = sys.argv[2]

# ... 导入模块后获取类
ModelClass = getattr(model_module, target_class)

运行命令就变成：

python3 script.py model_1 MyModel

注意事项 #

• 如果模型模块不在主脚本同一目录，可以通过sys.path.append("模块所在目录路径")把路径加入Python的搜索路径；
• 一定要加上异常处理，避免因为参数错误、模块缺失或类名错误导致脚本直接崩溃；
• 确保你的模型模块符合Python模块规范（比如文件名是合法的标识符，没有特殊字符）。

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