关于python：如何将n×n numpy数组中的每个元素平方？


How can I square every element in an n by n numpy array?

本问题已经有最佳答案，请猛点这里访问。

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

相关讨论

如果A是numpy数组，则只需输入A * A。

正如K. Tom所建议的，您可以执行A * A，也可以执行A ** 2

1
2
3
4
5
6
import numpy as np

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

print array * array #[1 4 9]
print array ** 2    #[1 4 9]
相关讨论

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

1
2
3
4
5
6
7
8
9
10
11
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)