自定义Pandas子类切片后丢失继承特性的问题排查与解决

• Pandas里像query、切片这类返回新对象的方法，默认只会输出原生的pd.DataFrame——哪怕你一开始实例化的是自定义子类Mypandas，这些操作内部都会调用原生构造函数生成新对象，不会保留你的子类类型。所以执行df.query('count > 0')后，返回的已经是普通DataFrame了，自然找不到foo()方法。

要让Pandas操作后仍返回自定义子类实例，只需在Mypandas类里重写_constructor属性，告诉Pandas创建新对象时用你的自定义类。

修正后的完整代码 #

import pandas as pd

class Mypandas(pd.DataFrame):
    # 重写_constructor，指定新对象的构造类
    @property
    def _constructor(self):
        return Mypandas
    
    def foo(self):
        return self

测试验证 #

# 先创建一个普通DataFrame测试
raw_df = pd.DataFrame({'count': [1, -2, 3, 0]})
# 转为Mypandas实例
my_df = Mypandas(raw_df)
# 执行query过滤
filtered_df = my_df.query('count > 0')
# 查看类型，确认是Mypandas
print(type(filtered_df))  # 输出 <class '__main__.Mypandas'>
# 调用自定义方法，正常执行
filtered_df.foo()

额外说明 #

• 除了query，像loc、iloc、groupby聚合这类返回新DataFrame的操作，现在都会自动返回Mypandas实例，不用再手动转换。
• 如果你的自定义类需要额外初始化参数，可能还要重写__finalize__方法来保留参数，但只是添加简单自定义方法的话，重写_constructor就足够。

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