Как сделать бесконечное получение данных из массива?

Question

[pcdesign]

lst = ['A', 'B', 'C']

Если страница
page = 1, то получаем A
page = 2, то получаем B
page = 3, то получаем C
page = 4, то получаем А
и т.д.

Answer 1

[longclaps]

from itertools import cycle

for c in cycle(['A', 'B', 'C']):
    print(c)

или

from itertools import count

lst = ['A', 'B', 'C']
for i in count():
    print(i, lst[i % len(lst)])

Comments

[pcdesign]

Спасибо за ответ.
Если мне нужно построить функцию, которая в зависимости от страницы будет отдавать значение.

То как в этом случае? Использовать enumerate для получения индекса и break для выхода?

[pcdesign]

>>> from itertools import cycle                   
>>> page = 5
>>> for i, c in enumerate(cycle(['A', 'B', 'C'])):
...     if i == page:
...             print(c)
...             break
... 
C

Или есть более оптимальное решение?

[longclaps]

pcdesign,
print(lst[(page - 1) % len(lst)])

[pcdesign]

longclaps, Спасибо!