Что делает функция индекс в Python?

Метод index() в Python помогает найти значение индекса элемента в последовательности. Он выдает наименьший возможный индекс указанного элемента. Если указанный элемент не существует, то возвращается ValueError.

Синтаксис:

sequence.index(x[, start[, end]])

• sequence - это объект, элементы которого представляют собой некую индексированную последовательность (строка, список, кортеж). Каждый элемент последовательности имеет индекс - целое число, обозначающее его позицию в последовательности. Нумерация индексов начинается с 0 (нуля).
• x - искомый элемент последовательности, чей наименьший индекс будет возвращен.
• Необязательные аргументы start и end используются для ограничения поиска конкретной подпоследовательностью списка, путем задания среза. Возвращаемый индекс вычисляется относительно изначальной последовательности, а не среза, заданного start и end.

str_seq = 'aebcdef'  #Последовательность в виде строки

print(str_seq.index('e'))
# => 1
print(str_seq.index('e', 2))
# => 5 
print(str_seq.index('k'))
# => ValueError: substring not found


lst_seq = ['a', 'e', 'b', 'c', 'd', 'e', 'f']  #Последовательность в виде списка
print(lst_seq.index('b'))
# => 2
print(lst_seq.index('k'))
# => ValueError: 'k' is not in list

tup_seq = ('a', 'e', 'b', 'c', 'd', 'e', 'f')  #Последовательность в виде кортежа
print(tup_seq.index('a'))
# => 0

index() — метод списка, строки и кортежа. Он возвращает позицию первого вхождения элемента:

fruits = ['яблоко', 'груша', 'слива', 'груша']
print(fruits.index('груша'))  # 1

Индексы в Python начинаются с нуля. Следующие вхождения index() не видит.

Если элемента нет

fruits.index('банан')
# ValueError: 'банан' is not in list

Проверьте наличие заранее:

if 'банан' in fruits:
    print(fruits.index('банан'))
else:
    print('Нет в списке')

Поиск в части списка

Второй и третий аргументы задают, с какой позиции искать и до какой:

print(fruits.index('груша', 2))  # 3 — поиск с индекса 2

Все вхождения

positions = [i for i, f in enumerate(fruits) if f == 'груша']
print(positions)  # [1, 3]

index() у строки

text = 'hello world'
print(text.index('o'))      # 4
print(text.index('world'))  # 6

У строк есть find(). Он ведёт себя как index(), но при отсутствии подстроки возвращает -1 без ошибки:

print(text.find('python'))  # -1

У списков метода find() нет.

Индекс максимального элемента

numbers = [3, 9, 4]
print(numbers.index(max(numbers)))  # 1

Скорость

index() перебирает элементы по очереди. На списке из миллиона элементов и частых поисках это медленно. Если нужно часто находить позицию, постройте словарь один раз:

pos = {f: i for i, f in enumerate(fruits)}

В словаре останется последняя позиция повторяющихся элементов.

Индексы в pandas

В pandas «индекс» — подписи строк таблицы df.index. Это другое понятие, с методом списка index() его не путайте.

Списки и строки разбирают в бесплатном курсе «Основы Python».