numpy蒙面数组或布尔索引?

时间:2013-11-30 23:23:44

标签: python numpy

我想执行点(A,A.T),其中省略了某些索引:

In [6]: A = np.array([[1,2,3],[4,5,6],[7,8,9]])
In [6]: import numpy.ma as ma

In [7]: Ama = ma(A)
In [9]: Ama[0,0] = ma.masked
In [11]: ma.dot(Ama,Ama.T)
Out[11]: 
masked_array(data =
 [[13 28 43]
 [28 77 122]
 [43 122 194]],
         mask =
 [[False False False]
 [False False False]
 [False False False]],
   fill_value = 999999)

最快的方法是什么?请注意,我想保留原始尺寸,即如果A是正方形,则点积将具有相同的尺寸。

0 个答案:

没有答案