Python | 列表中的 K 取幂

在Python中，可以使用 ** 运算符进行幂的运算。假设我们有一个列表，我们想计算列表中的每个元素的 2 次幂，可以通过以下方式实现：

lst = [1, 2, 3, 4, 5]
power_lst = [x ** 2 for x in lst]
print(power_lst)

这将输出：

[1, 4, 9, 16, 25]

同样，如果我们想计算列表中每个元素的 3 次幂，只需将幂指数从 2 改为 3：

lst = [1, 2, 3, 4, 5]
power_lst = [x ** 3 for x in lst]
print(power_lst)

这将输出：

[1, 8, 27, 64, 125]

但这并不是我们想要的。我们希望在运行时指定要计算的幂次数。为了实现这一点，我们可以将幂次数作为函数参数传递。下面是一个实现：

def power_list(lst, k):
    return [x ** k for x in lst]

lst = [1, 2, 3, 4, 5]
power_lst = power_list(lst, 2)
print(power_lst)

这将输出：

[1, 4, 9, 16, 25]

如果我们要计算 3 次幂，将要幂次数改为 3 即可：

lst = [1, 2, 3, 4, 5]
power_lst = power_list(lst, 3)
print(power_lst)

这将输出：

[1, 8, 27, 64, 125]

通过这种方式，我们可以方便地计算列表中的元素的任意幂次数。