创建Pandas.DataFrame具有返回datafram的方法的包装器

2024-04-27 04:36:21 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图创建一个类,它是pandas.DataFrame对象的包装器。在我为这个类编写了自己的方法之后,我希望pandas的方法也可以使用,但是要明确地告诉我/用户他们的方法来自pandas。它会像这样工作

df = pd.DataFrame(np.random.randn(5,2))
md = myData(df)

a = md.df # returns the original pandas.DataFrame "df" to a (equivalent of a=df
print(md) # prints as myData class
print(md.df) # prints just as print(df) would. Equiv to print(df)

md.mean() # mean as defined in myData class. Returns myData object
md.df.mean() # mean as defined in pandas. Returns dataframe object

md.std() # myData std
md.df.std() # pandas std

到目前为止,我所做的一切尝试都失败了。有一件事我真的认为它应该但是没有

^{pr2}$

它以RuntimeError: maximum recursion depth exceeded while calling a Python object退出。在

编辑

以下代码以相同的错误结尾。在

import pandas as _pd
class myData(_pd.DataFrame):
    """
    Attempt to create a myData object
    """
    def __init__(self, df, dic):
        df = df.copy()
        self.dic = dic
        super(myData, self).__init__(df)
        self.df = df

但是,如果我尝试

    def __init__(self, df, dic):
        df = df.copy()
        super(myData, self).__init__(df)

然后它就可以工作了,但是结果是一个myData对象,它实际上是一个DataFrame,因为每个方法都已经是DataFrames的方法了。在

你知道代码可能有什么问题吗?或者有什么方法可以让它变得更好吗?在


Tags: 方法selfdataframepandasdfobjectinitas