python - 如何在 python 中切片进程 itertools.product？

Question

我想处理一个非常大的 itertools.product 对象。问题是这样的:

import string
from itertools import product
text = string.lowercase[:] + string.uppercase[:] + '0123456789'
items = product(text, repeat=5)
for item in items:
    #do something

我知道 items 的长度是 62**5。如果我想处理items中索引范围从300000到600000的元素，如何实现？

我尝试将 itertools.product 转换为 python 列表，如下所示:

items = list(product(text, repeat=5))[300000:600000+1]
for item in items:
    #do something

但是好像这个转换消耗了很多内存，因为我一直在等待这个转换，最后放弃了。

我有这个需求是因为我想在python gevent中做这件事情，所以我想将大的itertool.product切片成小的item，用于gevent spawn。

Answer 1

您可以使用 islice做这个。

from itertools import product, islice
import string 

text = string.ascii_letters + string.digits
for item in islice(product(text,repeat=5), 300000, 600000):
    # do something