python - 如何将生成器对象平均分配到多个列表

Question

我为不耗尽内存的数百万个项目创建了生成器，现在为了处理，我想将这些项目平均分配到多个列表中：

例如：示例代码：

from itertools import islice

batch_size = 500

def get_ob():
    for i in range(20000000):
        yield i


obj = get_ob()
x = []
counter = 1
while True:
    objects = list(islice(obj, batch_size))

    if not objects:
        break
    x.append(objects)

创建这些多个列表需要很长时间。有没有更好的方法来解决这个问题。请建议。

标签： python