enumerate в Python: индекс и значение вместе

Цикл из первого урока отдаёт элементы, но не их номера. А номера нужны постоянно: «1. Анна» в списке участников, «строка 3» в отчёте, «позиция 2» в чеке. Классическое решение — ручной счётчик: переменная, которую вы объявляете до цикла и увеличиваете внутри. Оно работает, но занимает три строки и оставляет три шанса ошибиться. В Python есть встроенная функция enumerate, которая отдаёт номер и значение вместе.

Ручной счётчик: работает, но шумит

Сначала — как нумеруют без enumerate, чтобы было с чем сравнивать:

i = 1
for name in ['Анна', 'Борис', 'Вера']:
    print(f'{i}. {name}')
    i = i + 1

1. Анна
2. Борис
3. Вера

Вывод верный, но цена высока. Три заботы: объявить i = 1 до цикла, не забыть i = i + 1 внутри и увеличить строго в нужном месте — строка инкремента, потерявшаяся в длинном теле, ломает всю нумерацию. Забытый счётчик даст одинаковые номера на каждой строке. Всё это — ручная бухгалтерия вокруг того, что Python умеет считать сам.

enumerate: номер и элемент парой

enumerate оборачивает последовательность и отдаёт циклу пары: номер и элемент. Два имени в заголовке распаковывают пару — номер в i, элемент в name:

names = ['Анна', 'Борис', 'Вера']

for i, name in enumerate(names):
    print(f'{i}. {name}')

0. Анна
1. Борис
2. Вера

Никаких объявлений до цикла и инкрементов внутри — нумерация живёт в заголовке. Обратите внимание: нумерация начинается с нуля, как индексы списка: Анна — элемент с номером 0. То есть i здесь — это честный индекс: по нему можно обращаться к элементу, он совпадает с позициями срезов и range. enumerate отдаёт пары: номер и элемент вместе, и нумерация по умолчанию идёт с нуля.

start=1: нумерация с единицы

Для людей нумерация с нуля непривычна: списки покупок и чеки начинаются с единицы. Аргумент start задаёт, с какого номера начинать отсчёт:

names = ['Анна', 'Борис', 'Вера']

for num, name in enumerate(names, start=1):
    print(f'{num}. {name}')

1. Анна
2. Борис
3. Вера

Порядок перебора не изменился — элементы идут в прежней последовательности. start влияет только на числа, которые видит ваша печать: первый элемент получит номер 1 вместо 0. Сам список от этого не двигается, индексы остаются 0, 1, 2 — запомним это для питфолла ниже.

Что отдаёт enumerate без распаковки

Чтобы увидеть механизм, напечатаем пары целиком, без распаковки в две переменные:

names = ['Анна', 'Борис']

for pair in enumerate(names):
    print(pair)

(0, 'Анна')
(1, 'Борис')

Каждый шаг отдаёт кортеж из двух элементов: (0, 'Анна') — номер и значение. Распаковка for i, name in ... лишь раскладывает этот кортеж по двум переменным, как мы делали с items() в уроке про перебор словаря. Скобки в печати — подсказка: если в переменной кортеж, а вы ждали число, значит распаковки нет.

Номер, который начинается не с нуля

Аргумент start умеет больше, чем просто сдвинуть счёт на единицу. В него можно положить любое целое число, и отсчёт пойдёт от него: так строят нумерацию с конца, с обратного знака или с произвольного смещения.

for num, ch in enumerate('Кот', start=-2):
    print(num, ch)

-2 К
-1 о
0 т

Три буквы получили номера -2, -1 и 0 — отсчёт идёт к нулю с конца. Так выглядит позиция символа относительно середины строки: у слова из трёх букв средняя буква имеет номер 0. Для обычных задач это редкость, но знать полезно: start — это просто начало отсчёта, а не флаг «с единицы».

Когда номер реально нужен

Соблазн не пользоваться enumerate велик: номер почти никогда не нужен, и его добавление выглядит лишним. Но есть случаи, где без номера не обойтись, и вот они встречаются постоянно. Первый — скидки по ступеням: цена падает на каждом шаге, и шаг должен быть виден в выводе.

price = 1500
discounts = [0, 100, 300]
for step, off in enumerate(discounts):
    print(f'шаг {step}: цена {price - off}')

шаг 0: цена 1500
шаг 1: цена 1400
шаг 2: цена 1200

Здесь номер — не украшение, а часть данных: он показывает, на каком шаге скидка применяется. Сравните с обычным циклом: без номера пришлось бы заводить счётчик вручную, а потом ещё объяснять, что он означает.

Второй случай — словарь позиций. Когда элементы имеют смысл только вместе со своим местом, индекс становится ключом:

words = ['кот', 'пароход', 'ёж']
positions = dict(enumerate(words))
print(positions)
print(positions[1])

{0: 'кот', 1: 'пароход', 2: 'ёж'}
пароход

Получился словарь «номер — слово». Теперь по номеру можно достать элемент за одно обращение, без поиска по списку. Тот же приём с двумя списками разбирается в уроке про zip — там пары склеиваются, а здесь нумеруются.

Нумерованный чек: распаковка в два этажа

Если элементы сами являются парами — например, товары с ценами, — распаковка вкладывается: два имени для enumerate и скобки для вложенной пары:

cart = [('Хлеб', 42), ('Молоко', 89)]

for num, (name, price) in enumerate(cart, start=1):
    print(f'{num}. {name} — {price} р.')

1. Хлеб — 42 р.
2. Молоко — 89 р.

num получает позицию, скобки (name, price) распаковывают вложенный кортеж товара. Так собираются настоящие отчёты: нумерованные списки, построчные чеки, оглавления.

names = ['Анна', 'Борис', 'Вера']

for num, name in enumerate(names, start=1):
    print(names[num])

Нумерация и пары: enumerate вместе с zip

Две задачи — «нужен номер» и «нужны пары из двух списков» — обычно встречаются вместе: пронумерованный чек, отчёт с колонкой позиции, таблица с порядковым номером. Складывать enumerate и zip можно напрямую, но порядок имён в заголовке меняется, и это место, где новички спотыкаются чаще всего.

names = ['Аня', 'Борис']
scores = [82, 91]
for i, (name, score) in enumerate(zip(names, scores), start=1):
    print(i, name, score)

1 Аня 82
2 Борис 91

Читается справа налево: сначала zip склеивает имена с баллами в пары, потом enumerate нумерует эти пары, а распаковка в скобках разбирает каждую пару на два значения. Скобки обязательны — без них Python попытается распаковать в i и name сразу два элемента, а это уже не пара, а всё вместе.

enumerate по строке и до него: range(len())

enumerate работает по любой последовательности — и по строке тоже: номера достанутся символам:

for i, ch in enumerate('Кот'):
    print(i, ch)

0 К
1 о
2 т

А вот как в старом стиле нумеруют через диапазон: range(len(names)) даёт индексы, и элемент достаётся по индексу из списка:

names = ['Анна', 'Борис', 'Вера']

for i in range(len(names)):
    print(f'{i + 1}. {names[i]}')

1. Анна
2. Борис
3. Вера

Результат тот же, что у enumerate с start=1, — но обратите внимание на цену: i + 1 вручную и names[i] вместо имени элемента. Такой код встречается в чужих программах, читать его полезно, писать — уже не обязательно: enumerate короче и яснее. Диапазоны и их границы разобраны в уроке про range.

• нумерованные отчёты и списки: enumerate(..., start=1) даёт человеческую нумерацию;
• позиции в данных: где встретился нужный символ, на какой строке файла ошибка — номера нужны и при отладке;
• выравнивание номеров в колонках делают f-строки с шириной — формат-спецификации f-строк показывают как;
• когда номер не нужен, а элемент есть — enumerate лишний: обычный for из первого урока делает ту же работу короче.

Теперь цикл умеет и данные, и счёт. Следующий урок добавляет новый вид цикла — while: повторять не «по элементам», а «пока верно условие», когда число шагов заранее неизвестно.