Python中如何单行导入库的所有子模块，避免多行导入语句冗余？

报错原因解释 #

pytorch_forecasting的顶层__init__.py默认没有自动导入深层级的子模块，所以你直接执行import pytorch_forecasting as torchf时，只有顶层模块被加载，data模块下的examples子模块没有被挂载到父模块的属性上，直接访问就会抛出AttributeError。你之前写的from pytorch_forecasting.data.examples import get_stallion_data相当于手动触发了子模块的加载，所以之后访问torchf.data.examples才会生效。

实现极简导入、支持层级调用的方案 #

方案1：一行导入全量子模块（最符合你的需求）

借助Python标准库的importlib和pkgutil，用一行代码递归导入目标库的所有子模块，导入完成后即可直接通过torchf.xxx.yyy.zzz的格式调用任意层级的接口，完全不会出现属性找不到的问题，也不会污染全局命名空间：

# 同时导入pytorch_forecasting和pytorch_lightning所有子模块
import importlib, pkgutil, pytorch_forecasting as torchf, pytorch_lightning as torchl; [importlib.import_module(name) for lib in [torchf, torchl] for _, name, is_pkg in pkgutil.walk_packages(lib.__path__, f"{lib.__name__}.")]

方案2：轻量按需导入子模块

如果不想导入全量模块增加启动开销，也可以只导入你用到的子模块，比逐行导入单个类的写法简洁很多：

import pytorch_forecasting as torchf, pytorch_lightning as torchl
from pytorch_forecasting import data, metrics, models, Baseline, TemporalFusionTransformer, TimeSeriesDataSet
from pytorch_lightning import callbacks, loggers

调用示例 #

两种方案导入完成后都可以直接按你想要的格式调用，不需要额外写导入语句：

x = torchf.data.examples.get_stallion_data()
normalizer = torchf.data.GroupNormalizer()
smape = torchf.metrics.SMAPE()
early_stop = torchl.callbacks.EarlyStopping()

方案优势 #

• 完全满足你避免命名冲突的要求：所有接口都通过库别名层级调用，不会把类/函数名导入到全局命名空间，不会出现不同库同名接口覆盖的问题
• 导入语句极简，不需要逐行声明每个要用到的类/函数
• 不需要修改库本身的源码，也不需要额外安装第三方依赖

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