当给定列表和整数时，返回列表中的元素

Question

我正在为一个家庭作业问题写一个函数。该函数接受两个参数——一个列表和一个整数。输入列表和整数时，函数应返回列表中的元素，该元素具有与整数对应的索引

这是我写的一些代码：

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

当我运行这段代码时，我不断得到一个语法错误，说“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()是一个内置函数，它为类似lst的iterable中的每个元素返回一个2元组(element, index)