Python-查找列表的最后一个元素

Question

我正在codingbat.com上进行一些练习。我坚持的练习是这样的：

“给定一个int数组，如果6作为数组的第一个或最后一个元素出现，则返回True。该数组的长度为1或更大。

first_last6([1, 2, 6]) ? True

first_last6([6, 1, 2, 3]) ? True

first_last6([13, 6, 1, 2, 3]) ? False"

这是我要回答的内容。我知道这是错误的，但是我不确定自己做错了什么。

def first_last6(nums):
    if nums[0] == 6 or nums[len(nums)-1] == 6:
       return True
    else:
       return False

Answer 1

您的解决方案将真正起作用。这是一个稍微好一点的：

def first_last6(nums):
    return nums[0] == 6 or nums[-1] == 6