迭代给定父类中子类的某些方法

Question

我构建了两个名为 zero_coupon_bond 和 coupon_bond 的类，它们都有一个名为 Create_random_asset 的方法，它为两种产品生成随机数据属性（notionals、start_date、end_date、Riskdate 等）并在数据框中返回该数据（带有 n行）。现在我想写一个名为 product_collection，它继承了两个类并迭代了两个子类（会有更多的产品，这就是为什么我想用迭代来做），并使用方法 create_random_asset 为每个子类返回这些数据帧。在代码中：

class zero_coupon_bond():

    def __init__(self):
        ..........


    def create_random_asset(self,n):
        .....
        return pd.DataFrame(OrderedDict(self.columns))

class coupon_bond():

    def __init__(self):
        ..........

    def create_random_asset(self,n):
        .....
        return pd.DataFrame(OrderedDict(self.columns))

然后我正在寻找类似的东西：

class product_collection(zero_coupon_bond,coupon_bond):

    def __init__(self,n,k):
        self.n=n
        self.k=k
        self.df="empty dateframe"


    def random_data(self,n,amount_products):

               for product in childclass:
                    df.append(childclass.create_random_asset(n))

任何人知道我该如何正确实现它？提前致谢。

Answer 1

根据您在问题中提供的内容，您的 product_collection 类不应继承（也不是其祖先）您的优惠券类。相反，它应该尽可能地与coupon_bond 类解耦。也就是说，它应该能够识别和迭代任何实现方法create_random_asset的对象。

product_collection 类可以管理其元素类型为 coupon_bond 的内部集合（可能是 list）。

那么你的random_data 方法可以这样实现：

class product_collection():
    def __init__(self):
        self.products = []

    # ...

    def random_data(self):
        df = # empty dataframe
        for p in self.products:
            df.append(p.create_random_asset())
        return df

要将不同的产品添加到您的产品集合中，您可以在 product_collection 中实现一个 add_product 方法，该方法只是将一个 coupon_bond 实例附加到产品的内部列表中。

def add_prodct(self, product):
    assert product
    self.products.append(product)