python - 从不可下标的可迭代对象中获取第 n 个元素的更好方法

Question

有时一个可迭代对象可能是不可下标的。说返回itertools.permutations：

ps = permutations(range(10), 10)
print ps[1000]

Python会抱怨'itertools.permutations' object is not subscriptable

当然，可以按次执行next()以n获得第 n 个元素。只是想知道有没有更好的方法呢？

Answer 1

只需使用nth配方itertools

>>> from itertools import permutations, islice
>>> def nth(iterable, n, default=None):
        "Returns the nth item or a default value"
        return next(islice(iterable, n, None), default)

>>> print nth(permutations(range(10), 10), 1000)
(0, 1, 2, 4, 6, 5, 8, 9, 3, 7)

Answer 2

一个更具可读性的解决方案是：

next(x for i,x in enumerate(ps) if i==1000)