Python中如何使用if语句实现模块的条件导入功能

Python完全支持你描述的条件导入需求，你给出的示例代码本身就是合法可运行的，针对生产环境的稳定性要求，建议做如下优化：

• 替换两个独立的if判断为if-elif-else结构，从逻辑上保证只会触发一个导入分支，彻底避免参数异常导致同时导入两个模块的问题
• 补充参数合法性校验，避免传入非0/1值时两个模块都未导入，后续调用时出现难以排查的NameError
• 注意导入的作用域：如果条件导入逻辑写在函数内部，导入的模块别名默认仅在函数内部生效，如需全局使用需要显式声明对应别名为全局变量

优化后的示例代码如下：

# 参数前置校验
if parameter not in (0, 1):
    raise ValueError(f"parameter仅支持传入0或1，当前值为{parameter}")

if parameter == 0:
    # 如需全局使用np别名，显式声明全局
    global np
    import numpy as np
elif parameter == 1:
    global pd
    import pandas as pd

如果后续业务逻辑中对两个模块的调用方法兼容，还可以统一模块别名，减少后续代码的分支判断：

if parameter == 0:
    import numpy as common_lib
elif parameter == 1:
    import pandas as common_lib

# 后续代码直接使用common_lib调用对应方法即可

如果需要更灵活的动态导入（比如模块名由配置动态生成），也可以使用Python标准库的importlib.import_module方法实现，不过针对你当前固定两个模块的场景，直接用条件分支写导入语句可读性更高，更适合生产环境维护。

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