python - 错误 :list index out of range

Question

我是 Python 和一般编程的初学者，我正在尝试解决 codingbat.com 上的一个问题(该问题在“Warmup-2”部分下称为“array_front9”)。

问题是:“给定一个整数数组，如果数组中的前 4 个元素之一是 9，则返回 True。数组长度可能小于 4。”

这是我的代码，如果我创建一个列表然后在本地运行它(在 codingbat.com 上有必要创建一个函数，但我没有创建一个函数来在本地测试我的代码):

arr = [14,9,28,55,66,33,789,4548]

    for i in range (4):

        if arr[i] == 9:

            print('True')

    print('False')*

这是我尝试在 codingbat.com 上运行的代码，但我收到“错误:列表索引超出范围”错误:

def array_front9(arr):

    for i in range(4):

        if arr[i] == 9:

            return True

     return False

这是根据 codingbat.com 的解决方案:

def array_front9(nums):
  # First figure the end for the loop
      end = len(nums)
          if end > 4:
              end = 4

for i in range(end):  # loop over index [0, 1, 2, 3]
    if nums[i] == 9:
      return True
    return False

这是此问题的当前 URL:http://codingbat.com/prob/p110166

谁能指出我做错了什么？

Answer 1

题目中给出数组长度可能小于4，你只是在遍历一个数组。您应该首先检查数组 是否至少为length 4。如果 array 小于 4，则仅遍历 array 的长度。

n = 4

if len(arr) < 4:
    n = len(arr)

for i in range(n):
    #...Your code...