Поиск индекса элемента списка: примеры кода на Python

Расширенные примеры показывают менее очевидные возможности поиска индекса в списке.

Поиск с отрицательными границами. Параметры start и end принимают отрицательные значения, как при обычной работе со срезами. Отрицательное число обозначает позицию от конца списка.

vals = [10, 20, 30, 40, 50]
print(vals.index(40, -2))   # старт с индекса 3
print(vals.index(30, 1, -1)) # от индекса 1 до предпоследнего

3
2

Поиск кортежа по содержимому. Метод index работает с любыми сравнимыми объектами. Кортежи считаются равными, если их элементы совпадают.

points = [(1, 2), (3, 4), (5, 6)]
idx = points.index((3, 4))
print(idx)

1

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

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

    def __eq__(self, other):
        return self.salary == other.salary

workers = [
    Worker("Anna", 60000),
    Worker("Ivan", 50000),
    Worker("Petr", 70000),
]
candidate = Worker("Elena", 60000)
print(workers.index(candidate))

0

Условный поиск с генератором. Функция next в сочетании с enumerate находит первый элемент по произвольному условию и возвращает индекс. Второй аргумент next служит значением по умолчанию, если подходящего элемента нет.

people = [
    {"name": "Анна", "age": 25},
    {"name": "Игорь", "age": 19},
    {"name": "Ольга", "age": 31},
]
idx = next((i for i, person in enumerate(people) if person["age"] >= 30), -1)
print(idx)

2

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

events = ["click", "scroll", "click", "hover", "click"]
event = "click"
start = 0
indices = []
while True:
    try:
        found = events.index(event, start)
        indices.append(found)
        start = found + 1
    except ValueError:
        break
print(indices)

[0, 2, 4]

Индекс максимума или минимума. Метод index позволяет найти позицию максимального или минимального значения в списке. Одновременный вызов max и min делает код читаемым.

prices = [150, 90, 230, 175]
print(prices.index(max(prices)))
print(prices.index(min(prices)))

2
1

Преобразование индекса в отрицательный. Иногда удобно представить позицию как расстояние от конца. Для этого из индекса вычитают длину списка.

letters = ["a", "b", "c", "d"]
idx = letters.index("c")
print(idx)
print(idx - len(letters))

2
-2

Поиск в сортированном списке без index. Для отсортированного списка операция бинарного поиска находит позицию за логарифмическое время. Модуль bisect возвращает точку вставки, после чего проверяется совпадение значения.

from bisect import bisect_left

sorted_items = [10, 20, 30, 40, 50]
target = 30
pos = bisect_left(sorted_items, target)
if pos < len(sorted_items) and sorted_items[pos] == target:
    result = pos
else:
    result = -1
print(result)

2

Сложности сравнения булевых значений и чисел. Оператор == считает True равным 1, поэтому метод index может вернуть неожиданную позицию. Список может содержать 0, False и True одновременно, и индекс определяется первым совпадением по правилам равенства.

flags = [0, False, 2, True]
print(flags.index(True))
print(flags.index(1))

3
3

Список с условием и перебор всех подходящих индексов. Генератор списка с enumerate применяется для любых проверок, а не только для равенства значению.

values = [-5, 3, -2, 7, -1]
positive_indexes = [i for i, v in enumerate(values) if v > 0]
print(positive_indexes)

[1, 3]