如何继承pandas DataFrame类并通过文件路径实例化？遇递归错误求助

错误原因分析 #

你遇到的最大递归深度错误是由于__new__和__init__的循环调用导致的：

• 执行mc = my_class(save_path)时，Python会先调用__new__创建类实例，随后自动调用__init__初始化该实例。
• 你的__init__方法中又调用了self.__new__(pd.read_csv(save_path))，这会再次触发__new__创建新实例，进而再次调用__init__，形成无限循环。

正确实现方式 #

要实现通过CSV路径直接实例化自定义DataFrame子类，需要在__new__方法中完成CSV读取与实例创建，避免递归调用。以下是可行的代码：

import pandas as pd

class MyClass(pd.DataFrame):
    def __new__(cls, save_path, *args, **kwargs):
        # 读取CSV文件生成DataFrame
        df = pd.read_csv(save_path, *args, **kwargs)
        # 将DataFrame转换为当前子类的实例
        instance = super().__new__(cls, df)
        return instance
    
    def __init__(self, save_path, *args, **kwargs):
        # 调用父类DataFrame的初始化方法
        super().__init__()
        # 自定义属性赋值
        self.save_path = save_path
    
    def my_func(self):
        print('New function!')

# 测试代码
save_path = 'test_my_class.csv'
# 生成测试CSV
pd.DataFrame({'test_col': [1,2,3]}).to_csv(save_path, index=False)

# 实例化自定义类
mc = MyClass(save_path)
print(mc)
mc.my_func()
print(f"CSV路径: {mc.save_path}")

代码说明 #

1. __new__方法：作为类的构造方法，负责创建实例。先读取CSV得到普通DataFrame，再通过super().__new__(cls, df)将其转换为自定义子类的实例，确保实例继承DataFrame的所有特性。
2. __init__方法：负责初始化自定义属性（如save_path），调用super().__init__()确保父类DataFrame的初始化逻辑正常执行。
3. 支持传递pd.read_csv的额外参数（如sep、header等），增强灵活性。

另一种简洁实现方式 #

如果不需要额外的初始化逻辑，也可以用类方法来实现实例创建，代码更简洁：

import pandas as pd

class MyClass(pd.DataFrame):
    @classmethod
    def from_csv(cls, save_path, *args, **kwargs):
        df = pd.read_csv(save_path, *args, **kwargs)
        return cls(df)
    
    def my_func(self):
        print('New function!')

# 实例化方式
mc = MyClass.from_csv(save_path)

这种方式通过类方法封装CSV读取逻辑，避免重写__new__和__init__可能带来的复杂度。

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