如何实现Pandas层级式连续自定义访问器？

当然可以实现这种层级化的连续访问器，核心思路是利用Pandas访问器的实例化机制，在顶层访问器类中嵌套子访问器类，并让所有层级的访问器都持有原DataFrame的引用。

先搞懂Pandas访问器的基本机制 #

当你用@pd.api.extensions.register_dataframe_accessor("xxx")装饰一个类时：

• 这个类会被绑定到DataFrame的xxx属性上
• 当你调用df.xxx时，Pandas会自动创建该类的实例，并把当前的DataFrame对象传入实例的_obj属性（需要在类的__init__方法中接收并保存这个对象）
• 访问器类里的方法可以通过self._obj操作原DataFrame

层级化访问器的实现方式 #

要实现df.my_accessor_1.my_accessor_2...的链式调用，只需要在顶层访问器类中定义子访问器的属性，这些属性是子访问器类的实例，并且子访问器同样持有原DataFrame的引用。

代码示例

import pandas as pd

# 定义子访问器类（无需单独注册）
class MyAccessor2:
    def __init__(self, df):
        self._obj = df  # 保存原DataFrame引用
    
    def filter_by_value(self, column, value):
        """筛选指定列等于某个值的行"""
        return self._obj[self._obj[column] == value]

# 定义顶层访问器类，注册到DataFrame
@pd.api.extensions.register_dataframe_accessor("my_accessor_1")
class MyAccessor1:
    def __init__(self, df):
        self._obj = df
        # 初始化子访问器，传入原DataFrame
        self.my_accessor_2 = MyAccessor2(df)
    
    def get_shape_info(self):
        """返回DataFrame的形状信息"""
        return f"行数: {self._obj.shape[0]}, 列数: {self._obj.shape[1]}"

# 测试调用
if __name__ == "__main__":
    df = pd.DataFrame({"a": [1,2,3,4], "b": [5,6,7,8]})
    # 调用顶层访问器的方法
    print(df.my_accessor_1.get_shape_info())
    # 链式调用子访问器的方法
    filtered_df = df.my_accessor_1.my_accessor_2.filter_by_value("a", 2)
    print(filtered_df)

扩展说明 #

• 可以无限嵌套层级：只要在子访问器类中继续定义下一级的访问器属性即可
• 所有层级的访问器共享同一个原DataFrame引用，操作都是基于原始数据（如果需要修改数据，注意是否要返回新的DataFrame）
• 子访问器不需要单独用register_dataframe_accessor注册，它们是通过顶层访问器的属性暴露给外部的

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