python - 在 Python 中对数字列表(或数组？)进行平方

<分区>

我来自 MATLAB 背景，根据其他堆栈中的答案，到目前为止，这个简单的操作在 Python 中实现似乎非常复杂。通常，大多数答案都使用 for 循环。

到目前为止我看到的最好的是

import numpy
start_list = [5, 3, 1, 2, 4]
b = list(numpy.array(start_list)**2)

有没有更简单的方法？

最佳答案

_{注意:因为我们已经有了 vanilla Python, list comprehensions and map 的副本而且我还没有找到一个重复项来对 1D numpy 数组进行平方，我想我会保留使用 numpy}

的原始答案

`numpy.square()`

如果您是从 MATLAB 转到 Python，那么为此尝试使用 numpy 绝对是正确的。使用 numpy 你可以使用 numpy.square()返回输入的元素平方:

>>> import numpy as np
>>> start_list = [5, 3, 1, 2, 4]
>>> np.square(start_list)
array([25,  9,  1,  4, 16])

还有一个更通用的numpy.power()

>>> np.power(start_list, 2)
array([25,  9,  1,  4, 16])

关于python - 在 Python 中对数字列表(或数组？)进行平方，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/35696811/