numpy掩码数组还是布尔索引?

2024-06-02 07:52:00 发布

您现在位置:Python中文网/ 问答频道 /正文

我想执行dot(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是正方形,则点积将具有相同的尺寸。在


Tags: inimportnumpyfalse尺寸asnpout