将未定义的方法调用传递给包含不同对象的属性

Question

我有一个属性为pandas DataFrame()的类.我想设置类,如果在其上调用任何DataFrame()方法,该方法将应用于DataFrame().

我已经设置了这个场景:

import pandas as pd
class A(object):
    def __init__(self):
        self.df = pd.DataFrame()
        self.name = 'Charlie'

    def get_name(self):
        return self.name

    def head(n=5):
        return self.df.head(n)

所以我可以这样做:

a = A()
a.get_name()

出[1]:'查理'

a.head()

Out [2]:空DataFrame()

我不想重新定义类的每个方法都要通过.这只是一个例子.我想把它应用到其他地方.

谢谢,

皮

Answer 1

您需要覆盖__getattr__类的方法A并委托所有调用self.df,如果找不到该属性self.df,则在当前对象中查找它.我的系统上没有Pandas,所以我将为演示定义另一个类:

class B(object):

    def func(self):
        print 'Inside B'

class A(object):

    def __init__(self):
        self.df = B()
        self.name = 'Charlie'

    def get_name(self):
        return self.name

    def __getattr__(self, attr):
        try:
            return getattr(self.df, attr)
        except AttributeError:
            #If attribute was not found in self.df, then try in self
            return object.__getattr__(self, attr)

输出:

>>> a = A()
>>> a.func()
Inside B
>>> a.get_name()
'Charlie'