加入两个类对象 - Python答案

【问题标题】：Join two class objects - Python加入两个类对象 - Python
【发布时间】：2020-02-26 15:36:21
【问题描述】：

我有两个生成器：

generator1 = CustomGen(X1,X1, length=W, stride = 1,sampling_rate=1, batch_size=bt_size)
generator2 = CustomGen(X2,X2, length=W, stride = 1,sampling_rate=1, batch_size=bt_size)

使用Custom Data Generator for Keras LSTM with TimeSeriesGenerator中采用的函数创建的：

class CustomGen(TimeseriesGenerator):
    def __getitem__(self, idx):
        x, y = super().__getitem__(idx)
        # do processing here
        return x, x

每个对象有 m 个批次。我想将这两个对象合并为一个总共有 2m 个批次的对象。

【问题讨论】：

标签： python generator

【解决方案1】：

itertools.chain 可能是你的朋友：

generator = itertools.chain(generator1, generator2)

它将返回来自generator1 的所有元素，然后是来自generator2 的元素。

【讨论】：

感谢您的回复。 chain 的输出与 generator1（或 generator2）的类型不同。是否有保留对象类型的替代方法。