如何对n乘n阵列中的每个元素进行平方？

Question

我不想要任何for循环，并且想知道是否有我可以使用的功能。

Answer 1

如果A是numpy数组，我只需输入A * A.

Answer 2

正如K. Tom建议您可以A * A，您也可以A ** 2

import numpy as np

array = np.array([1,2,3])

print array * array #[1 4 9]
print array ** 2    #[1 4 9]

Answer 3

您可以使用np.square或np.power：

l =  [[1,2,3], [2,3,4]]

In [5]: np.power(l, 2)
Out[5]:
array([[ 1,  4,  9],
       [ 4,  9, 16]], dtype=int32)

In [6]: np.square(l)
Out[6]:
array([[ 1,  4,  9],
       [ 4,  9, 16]], dtype=int32)