方法链派生数据框类

Question

我想通过下面的代码

class CustomFrame(DataFrame): 
    def __init__(self): 
     DataFrame.__init__(self,{"a":[1,2,3,4],"b":[5,6,7,8]}) 
    def get(self): 
     return self.loc[1] 
    def foo(self): 
     return 2*self 

现在我想以扩展大熊猫数据框链的方法调用，如：

>>> c=CustomFrame() 
>>> c.get().foo() 
'Series' object has no attribute 'foo' 

明显.loc返回一系列哪些不知道有关自定义foo方法的任何信息。有什么办法可以做到这一点？

编辑：

理想我想这样做

c.method_from_data_frame().method_from_custom_frame().another_method_from_data_frame() 

继我已经覆盖的DataFrame构造HYRY的意见，但是我需要设置初始化一些属性CustomFrame

class CustomFrame(DataFrame): 
    def __init__(self, *args, **kw): 
     super(CustomFrame, self).__init__(*args, **kw) 
     self.c = kw.get('c',False) 

    @property 
    def _constructor(self): 
     return CustomFrame 

    >>> c=CustomFrame(c=5) 
    >>> print c.c 
    5 
    >>> print c.get().c 
    False 

我试过用functoolspartial

@property 
    def _constructor(self): 
     return partial(CustomFrame,c=5) 

但我得到一个pandas.core.common.PandasError: DataFrame constructor not properly called!错误。另一种方法是改变get到

def get(self): 
     ret = self.loc[[1]] # CustomFrame with default .c 
     ret.c = self.c 
     return ret 

这似乎不是很优雅

Answer 1

您可以覆盖_constructor财产，这里有一个例子，因为self.loc[1]返回一系列对象，我把它改为self.loc[[1]]：

from pandas import DataFrame 


class CustomFrame(DataFrame): 
    def __init__(self, *args, **kw): 
     super(CustomFrame, self).__init__(*args, **kw) 

    @property 
    def _constructor(self): 
     return CustomFrame   

    def get(self): 
     return self.loc[[1]] 

    def foo(self): 
     return 2*self 

c=CustomFrame({"a":[1,2,3,4],"b":[5,6,7,8]}) 
print c.get().foo() 

编辑

我没有更好的办法来复制的属性，也许你可以使用_metadata或装饰，这里是装饰的例子：

def copy_attrs(func): 
    def wrap_func(self, *args, **kw): 
     res = func(self, *args, **kw) 
     res.c = self.c 
     return res 
    return wrap_func 

class CustomFrame(DataFrame): 
    def __init__(self, *args, **kw): 
     self.__dict__["c"] = kw.pop("c", None)  
     super(CustomFrame, self).__init__(*args, **kw) 

    @property 
    def _constructor(self): 
     return CustomFrame 

    @copy_attrs 
    def get(self): 
     return self.loc[:2] 

df = CustomFrame({"a":[1,2,3,4],"b":[5,6,7,8]}, c=100) 
print df.c, df.get().c