如何让继承pandas.DataFrame的子类调用基类方法时返回子类实例

pandas 官方为 DataFrame 子类扩展预留了专用机制，不需要重写任何基类方法即可实现所有操作返回子类实例。

核心实现 #

只需要在子类中定义 _constructor 属性，指向自身构造函数即可：

import pandas as pd

class Subclass(pd.DataFrame):
    # 定义构造器返回子类本身
    @property
    def _constructor(self):
        return Subclass

    def new_method(self):
        # 你的自定义业务逻辑
        return Subclass(self.transform(...))

原理解释 #

pandas 内部所有返回新 DataFrame 实例的公共方法（包括切片、过滤、合并、统计变换等），都会调用当前实例的 _constructor 方法生成返回对象，而非直接硬编码调用 pd.DataFrame。重写该属性后，所有基类方法的返回结果都会自动替换为子类实例，保留自定义方法。

带自定义元数据的扩展 #

如果你的子类需要携带自定义元数据，额外定义 _metadata 属性声明需要保留的字段即可：

class Subclass(pd.DataFrame):
    # 声明需要在实例拷贝时保留的自定义元数据
    _metadata = ["business_tag"]

    @property
    def _constructor(self):
        return Subclass

    def __init__(self, *args, business_tag=None, **kwargs):
        super().__init__(*args, **kwargs)
        # 自定义业务属性
        self.business_tag = business_tag

效果验证 #

# 初始化子类实例
df = Subclass({"col1": [1,2,3], "col2": [4,5,6]}, business_tag="test")
# 调用任意基类方法
filtered_df = df[df["col1"] > 1]
print(type(filtered_df)) # 输出 <class '__main__.Subclass'>
print(filtered_df.business_tag) # 输出 test
# 支持链式调用自定义方法
filtered_df.new_method()

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