Как вычислить индекс нужного объекта в структуре данных list

Дополнительные примеры расширяют работу с индексом и показывают нестандартные ситуации.

Пример 1. Нахождение всех вхождений методом index со стартовой позицией

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

def all_positions(items, value):
    result = []
    current = 0
    while True:
        try:
            found = items.index(value, current)
            result.append(found)
            current = found + 1
        except ValueError:
            break
    return result

numbers = [1, 2, 1, 3, 1]
print(all_positions(numbers, 1))

[0, 2, 4]

Цикл продолжается до тех пор, пока index находит значение. Исключение ValueError служит сигналом окончания перебора.

Пример 2. Индекс минимального и максимального элемента

Для поиска позиции min или max достаточно сначала получить значение функцией min или max, а затем передать его в index. Альтернативный способ использует range и функцию ключа.

values = [12, 0, 18, 4, 97]
min_position = values.index(min(values))
max_position = max(range(len(values)), key=values.__getitem__)
print('Минимум:', min_position)
print('Максимум:', max_position)

Минимум: 1
Максимум: 4

Функция max с key вызывает values.__getitem__ для каждого индекса. Результатом становится индекс элемента с наибольшим значением.

Пример 3. Поиск индекса в списке списков и кортежей

Если список содержит вложенные объекты, равенство проверяется по содержимому кортежа или вложенного списка.

matrix = [[1, 2], [3, 4], [5, 6]]
sub = [3, 4]
matrix_pos = matrix.index(sub)
print('Подсписок:', matrix_pos)

points = [(0, 5), (2, 8), (4, 9)]
current = (2, 8)
point_pos = points.index(current)
print('Кортеж:', point_pos)

Подсписок: 1
Кортеж: 1

Вложенный объект должен совпадать по порядку значений. Список [4, 3] не считается равным [3, 4].

Пример 4. Поиск индекса объекта в списке по его свойствам

class Worker:
    def __init__(self, name, salary):
        self.name = name
        self.salary = salary

team = [
    Worker('Иван', 45000),
    Worker('Мария', 65000),
    Worker('Олег', 35000)
]
premium = next(
    (i for i, worker in enumerate(team) if worker.salary > 50000),
    -1
)
print(premium)

1

Метод index не подходит для сравнения по отдельному полю объекта. Обход enumerate даёт индекс и объект одновременно, поэтому условие можно отнести к свойству worker.salary.

Пример 5. Индекс в отсортированном списке с помощью bisect

from bisect import bisect_left

def sorted_index(sorted_items, value):
    position = bisect_left(sorted_items, value)
    if position != len(sorted_items) and sorted_items[position] == value:
        return position
    raise ValueError('Значение не находится в списке')

marks = [3, 4, 4, 5, 5, 5]
print(sorted_index(marks, 5))

3

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

Пример 6. Индексы элементов, соответствующих нескольким условиям

data = [14, 3, 19, 27, 8, 22, 5]
selected = [i for i, value in enumerate(data) if value > 10 and value % 2 == 0]
print(selected)

[0, 5]

Из списка выбраны значения 14 и 22, так как они больше 10 и делятся на два. Генератор списка помещает в результат индексы 0 и 5.