连接两个类对象-Python

我有两台发电机：

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)

使用以下所采用的函数创建的：

---

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

itertools。chain

可能是您的朋友：

generator = itertools.chain(generator1, generator2)

---

它将返回

generator1

中的所有元素，然后返回

generator2

中的元素，谢谢您的回复。链的输出与generator1（或generator2）类型不同。是否有一种替代方法，因此保留对象类型。

generator = itertools.chain(generator1, generator2)