python - 查找列表中的第 n 次幂

Question

给定一个包含正数和非负数 N 的数组。您应该找到数组中索引为 N 的元素的 N 次方。如果 N 在数组之外，则返回 -1。

def index(array, n):
    selected_number = []
    if n > len(array):
        return -1
    else:
        selected_number.append(array[n])
        total = selected_number[0]
    return total ** n

这是我写的代码。它正确地执行了第二部分，但是当变量 n 大于数组列表时，输出不会像预期的那样输出 -1。这怎么行不通？

标签： pythonlist

Answer 1

这是一个单行：

result = arr[n]**n if n<len(arr) else -1