2016-04-01 68 views

回答

4

如果A是numpy數組,我只需輸入A * A。

3

由於K.湯姆建議你可以做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] 
+0

我絕對不知道!謝謝。 – Jobs

+1

唯一要注意的是類型。 'a = np.array([[100,200],[50,150]],dtype = np.uint8); a * a'會溢出。 – Reti43

1

你可以使用np.squarenp.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) 
相關問題