Соедините два объекта класса - Python

Question

У меня есть два генератора:

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)

, которые созданы с использованием функции, принятой из Генератор пользовательских данных для Keras LSTM с TimeSeriesGenerator :

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

Каждыйобъект имеет м количество пакетов. Я хотел бы объединить эти два объекта в один, имеющий в общей сложности 2 м пакетов.

Answer 1

itertools.chain может быть вашим другом здесь:

generator = itertools.chain(generator1, generator2)

Он вернет все элементы из generator1, за которыми следуют элементы из generator2.