python - itertools 使用列表进行计数

Question

我想将 itertools.count 元素作为列表索引传递，但会导致以下错误:

TypeError: list indices must be integers or slices, not itertools.count

尝试int(counter)也不起作用，导致

TypeError: int() argument must be a string, a bytes-like object or a number, not 'itertools.count'

from itertools import count

index = count()
l = [5,6,7,8,9,0]

while True:
    print(l[int(index)])

如何将计数作为列表索引传递？

Answer 1

您可以使用next从计数器获取下一个元素。

from itertools import count

index = count()
l = [5, 6, 7, 8, 9, 0]

while True:
    print(l[next(index)])

但是按照当前循环的结构方式，当计数器达到 6 时，最终会导致 IndexError，因为您的列表只有 6 个元素。

输出将如下所示:

5
6
7
8
9
0
Traceback (most recent call last):
  File "test.py", line 7, in <module>
    print(l[next(index)])
IndexError: list index out of range

如果您只想打印列表的元素，您可以使用 Python 更轻松地完成此操作，如下所示:

for element in l:
    print(element)