如何在Python类中为属性封装带方法的包装类？

要实现这种既可以直接返回原始DataFrame、又能通过.to_matrix访问处理结果的属性，核心是用包装类封装原始数据，同时重写字符串输出方法模拟原始DataFrame的打印效果，再添加自定义属性提供处理后的结果。

完整实现代码 #

import pandas as pd
import numpy as np

class SpecialClass:
    def __init__(self, df):
        self._original_df = df  # 存储原始DataFrame

    def __str__(self):
        # 让print()调用时输出原始DataFrame的字符串
        return str(self._original_df)
    
    def __repr__(self):
        # 交互式环境下直接输出原始DataFrame的表示
        return repr(self._original_df)
    
    @property
    def to_matrix(self):
        # 这里实现你的数据处理逻辑，比如转成numpy矩阵
        return self._original_df.to_numpy()

class MyMainClass:
    def __init__(self, x: pd.DataFrame, y: pd.DataFrame):
        # 用SpecialClass包装原始的x和y
        self.x = SpecialClass(x)
        self.y = SpecialClass(y)

验证调用效果 #

# 测试用例
x_df = pd.DataFrame({'a': [1,2,3], 'b': [4,5,6]})
y_df = pd.DataFrame({'c': [7,8,9], 'd': [10,11,12]})

mainclass = MyMainClass(x_df, y_df)
print(mainclass.x)  # 打印原始DataFrame
# 输出：
#    a  b
# 0  1  4
# 1  2  5
# 2  3  6

print(mainclass.x.to_matrix)  # 打印处理后的矩阵
# 输出：
# [[1 4]
#  [2 5]
#  [3 6]]

关键细节说明 #

• SpecialClass的作用：作为原始DataFrame的包装层，既保留了原始数据的访问能力，又扩展了.to_matrix属性。
• __str__和__repr__重写：这两个方法决定了对象被打印或在交互式环境中显示的内容，直接返回原始DataFrame的对应方法结果，让用户感觉mainclass.x就是原始DataFrame。
• @property装饰器：把to_matrix变成属性而非方法，符合你需要的mainclass.x.to_matrix调用方式，无需加括号。

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