Возвращать элемент в списке, когда ему дан список и целое число

Question

Я пишу функцию для домашнего задания. Функция принимает два аргумента - список и целое число. Когда вы вводите список и целое число, функция должна возвращать элемент списка, индекс которого соответствует целому числу.

Это код, который я написал:

def find_element (sample_list, sample_integer):
   for element in sample_list:
      if index of element == sample_integer:
         return element

КогдаЯ запускаю этот код, я получаю синтаксическую ошибку, которая говорит, что «элемент» не определен. Я продолжаю пытаться разобраться в этом, и мне нужна помощь. Любой вход?

Answer 1

Python (и почти любой другой язык, который включает массивы или списки) имеет встроенный механизм для извлечения элементов списка по индексу:

def find_element(lst, index):
    return lst[index]

Нет причин усложнять это, хотяесли вы должны считать по индексу, вы можете сделать что-то вроде этого:

def find_element(lst, sample_idx):
    for elem, idx in enumerate(lst):
        if idx == sample_idx:
            return elem

, где enumerate() - это встроенная функция, которая возвращает 2-кортеж (element, index) для каждогоэлемент в итерируемый как lst.