什么 python 内置 function 可以将一些 function 应用于列表中的每个元素？

Question

我有 function sqr(n)和 list array 。 是否有 python 内置 function 将sqr(n)应用于数组的每个元素（数组内部的元素发生变化）？ 还需要一些使用map() function 或其他内置函数的技巧。

当然，这可以使用map() function 来完成，但它会创建一个迭代器，并且你会得到一个新列表（旧列表不会改变）：

def sqr(n):
    return n ** 2

array = [1, 2, 3, 4, 5]
array = list(map(sqr, array)) # have to reassign name to new list but elements don't change in old one.
print(array) # output: [1, 4, 9, 16, 25]

我想要这样的东西：

def sqr(n):
    return n ** 2

array = [1, 2, 3, 4, 5]
some_built_in_function(sqr, array)
print(array) # output: [1, 4, 9, 16, 25]

Answer 1

没有内置的 function，但您可以使用map和切片分配来实现相同的结果。

array[:] = map(sqr, array)

（如果你需要一个表达式而不是一个语句，你可以使用 monstrosity

array.__setitem__(slice(None), map(sqr, array))

这基本上就是切片分配的实现方式。 )

Answer 2

没有内置function，但是自己写一个很容易：

def mapinplace(l, fun):
    for i, val in enumerate(l):
        l[i] = fun(val)

array = [1, 2, 3, 4, 5]
mapinplace(array, sqr)

Answer 3

由于您很可能对数字进行操作，因此我将发布强制性 numpy 解决方案。 Numpy 不是完全内置的，但它非常接近：例如，python 中的@运算符是为 numpy 制作的。

仅当输入是 numpy 数组而不是列表时，此解决方案才会就地：

np.square(x, out=x)

大多数元素方面的 numpy 操作可以像这样就地完成。

Answer 4

您可以按如下方式使用列表理解来实现此目的

array = [1, 2, 3, 4, 5]
array = [i**2 for i in array]
print(array)