2

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

到目前为止我见过的最好的是

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

有没有更简单的方法?

4

4 回答 4

7

注意:由于我们已经有vanilla Python、list comprehensions 和 map的副本,并且我还没有找到一个副本来平方一维 numpy array,所以我想我会保留使用的原始答案numpy

numpy.square()

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

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

numpy.power()

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

>>> np.power(start_list, 2)
array([25,  9,  1,  4, 16])
于 2016-02-29T11:03:40.020 回答
5

最易读的可能是列表推导:

start_list = [5, 3, 1, 2, 4]
b = [x**2 for x in start_list]

如果您是功能类型,您会喜欢map

b = map(lambda x: x**2, start_list)  # wrap with list() in Python3
于 2016-02-29T10:11:31.190 回答
5

只需使用列表理解:

start_list = [5, 3, 1, 2, 4]
squares = [x*x for x in start_list]

注意:作为一个小的优化,doingx*xx**2(或pow(x, 2)).

于 2016-02-29T10:11:42.427 回答
4

您可以使用列表理解:

[i**2 for i in start_list]

或者,如果您正在使用numpy,您可以使用tolist方法:

In [180]: (np.array(start_list)**2).tolist()
Out[180]: [25, 9, 1, 4, 16]

或者np.power

In [181]: np.power(start_list, 2)
Out[181]: array([25,  9,  1,  4, 16], dtype=int32)
于 2016-02-29T10:12:02.210 回答