如何对一个二维的numpy数组进行逐元素的平方或乘方?

8 浏览
0 Comments

如何对一个二维的numpy数组进行逐元素的平方或乘方?

我需要对一个二维numpy数组进行平方(逐元素),我尝试了以下代码:

import numpy as np
a = np.arange(4).reshape(2, 2)
print a^2, '\n'
print a*a

得到的结果是:

[[2 3]
 [0 1]]
[[0 1]
 [4 9]]

很明显,使用`a*a`这种记号可以得到我想要的结果,而不是`a^2`。

我想知道是否存在另一种记号来将一个numpy数组提升到2次方或N次方?而不是使用`a*a*a*..*a`的方式。

0