无需导入定义模块即可使用自定义Pandas扩展的方法

你的思路完全可行——通过包的__init__.py文件自动加载Pandas扩展模块，就能避免在每个使用模块中重复导入。具体实现步骤如下：

• 在source包的__init__.py文件中添加一行导入语句：

  import source.pd_extensions
  

• 之后在my_module或其他同包子模块中，只要导入source包（或其子模块），就会自动触发pd_extensions的加载，Pandas扩展会完成注册，无需再单独导入source.pd_extensions。

原理说明 #

Pandas的register_dataframe_accessor是模块级别的注册操作——只要包含注册逻辑的模块被成功导入一次，所有后续创建的DataFrame实例都能直接使用该扩展方法。通过包的__init__.py导入pd_extensions，相当于把扩展注册逻辑绑定到了包的加载流程中，只要包被访问就会自动执行注册。

验证示例 #

在my_module中可以这样使用：

# 只需导入source包（或其子模块）
import source
import pandas as pd

df = pd.DataFrame({'col1': [1,2,3]})
# 直接调用扩展方法，无需单独导入pd_extensions
df.your_extension_method()

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