Индексация и порядковый номер в последовательностях Python

Расширенные примеры работы с индексами

Использование enumerate с файлами и строками

При чтении текстового файла часто нумеруют строки. enumerate позволяет сделать это элегантно.

with open('example.txt', 'w', encoding='utf-8') as f:
    f.write('Первая строка\nВторая строка\nТретья строка')

with open('example.txt', 'r', encoding='utf-8') as f:
    for num, line in enumerate(f, start=1):
        print(f'{num}: {line.strip()}')

1: Первая строка
2: Вторая строка
3: Третья строка

Пояснение: файловый объект итерируется построчно, enumerate добавляет номер строки. Метод strip() удаляет символ перевода строки.

Получение индексов всех вхождений элемента

Метод index возвращает только первое вхождение. Чтобы найти все индексы, применяется цикл с передачей начальной позиции.

fruits = ['apple', 'banana', 'cherry', 'banana', 'date']
target = 'banana'
indices = []
start = 0
while True:
    try:
        idx = fruits.index(target, start)
        indices.append(idx)
        start = idx + 1
    except ValueError:
        break
print(f'Все индексы {target}: {indices}')

Все индексы banana: [1, 3]

Пояснение: параметр start задает, с какой позиции искать. После каждого найденного индекса продолжаем поиск со следующей позиции.

Обход матрицы (двумерного списка) с индексами строк и столбцов

Вложенный enumerate дает индексы строк и столбцов.

matrix = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]
for i, row in enumerate(matrix):
    for j, val in enumerate(row):
        print(f'Элемент [{i}][{j}] = {val}')

Элемент [0][0] = 1
Элемент [0][1] = 2
Элемент [0][2] = 3
Элемент [1][0] = 4
Элемент [1][1] = 5
Элемент [1][2] = 6
Элемент [2][0] = 7
Элемент [2][1] = 8
Элемент [2][2] = 9

Пояснение: внешний цикл дает индекс строки, внутренний - индекс столбца. Можно также использовать enumerate с двумя переменными.

Изменение начального индекса в зависимости от условия

В некоторых задачах нужно пропускать первые N элементов или начинать нумерацию с определенного места. enumerate со срезом.

data = ['a', 'b', 'c', 'd', 'e']
start_index = 2
for idx, val in enumerate(data[start_index:], start=start_index):
    print(f'Индекс {idx}: {val}')

Индекс 2: c
Индекс 3: d
Индекс 4: e

Пояснение: срез data[start_index:] создает подпоследовательность, а enumerate начинает нумерацию с заданного числа. Важно: исходные индексы сохраняются относительно оригинального списка.

Использование zip с enumerate для параллельного обхода

Когда нужно обойти несколько последовательностей одновременно и получить индекс.

names = ['Alice', 'Bob', 'Charlie']
scores = [85, 92, 78]
for idx, (name, score) in enumerate(zip(names, scores)):
    print(f'{idx+1}. {name} - {score}')

1. Alice - 85
2. Bob - 92
3. Charlie - 78

Пояснение: zip объединяет элементы в кортежи, enumerate нумерует эти кортежи.

Получение индекса элемента без цикла: метод bisect для отсортированных списков

Для отсортированных последовательностей можно использовать бинарный поиск из модуля bisect для нахождения позиции вставки, что эквивалентно индексу.

import bisect
sorted_list = [10, 20, 30, 40, 50]
value = 30
pos = bisect.bisect_left(sorted_list, value)
print(f'Индекс элемента {value}: {pos}')

Индекс элемента 30: 2

Пояснение: bisect_left возвращает индекс, по которому нужно вставить элемент, чтобы сохранить порядок. Если элемент существует, это его первый индекс. Подходит только для отсортированных данных.

Примечание: во всех примерах индексы начинаются с 0, если не указано иное. Для получения индексов с 1 используйте start=1.