如何避免pandas自定义访问器注册时触发的覆盖警告

问题根因 #

该警告触发的核心原因是register_dataframe_accessor装饰器每次执行时都会尝试给DataFrame注册同名属性，重复注册时就会抛出警告。之前的过滤规则仅匹配主运行文件中定义的GeoAccessor类，当代码作为模块被导入时，类的模块前缀发生变化，导致规则匹配失效。

解决方案 #

方案1：调整警告过滤规则 #

利用filterwarnings支持正则匹配的特性，替换固定模块前缀为通配符，适配所有运行场景：

import warnings
warnings.filterwarnings(
    "ignore",
    message=r"registration of accessor <class '.*GeoAccessor'> under name 'geo' for type <class 'pandas.core.frame.DataFrame'> is overriding a preexisting attribute with the same name."
)

方案2：从根源避免重复注册（更推荐） #

注册前先判断DataFrame是否已经存在geo属性，仅在未注册时执行注册逻辑，不需要修改警告规则即可彻底解决问题：

import pandas as pd
import numpy as np

# 仅当geo访问器不存在时执行注册
if not hasattr(pd.DataFrame, "geo"):
    @pd.api.extensions.register_dataframe_accessor("geo")
    class GeoAccessor:
        def __init__(self, pandas_obj):
            self._obj = pandas_obj

        @property
        def center(self):
            # 返回DataFrame存储坐标的地理中心点
            lat = self._obj.latitude
            lon = self._obj.longitude
            return (float(lon.mean()), float(lat.mean()))

if __name__ == "__main__":
    ds = pd.DataFrame({"longitude": np.linspace(0, 10),
                   "latitude": np.linspace(0, 20)})
    print(ds.geo.center)

该方案无论是单文件运行还是作为模块被导入，都不会再触发对应警告。

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