首页 文章

Python,Numpy:如何用给定的掩码选择numpy数组

提问于
浏览
1

例如,给定(10000,250)大小的numpy矩阵A.

>>>A.shape
(10000, 250)

和一个numpy掩码数组 m

>>>m = np.arange(0, A.shape[0], 3)
>>>m
([0, 3, 6, 9, ....., 9997])

这将选择 A 的所需列

>>>A[m]
>>>A[m].shape
(3333, 250)

但我的问题是 . 如何选择 A 的其余部分? A[([1, 2, 4, 5, 7, 8, ...., 9998, 9999, 10000])]

1 回答

  • 2

    您可以使用setdiff1d选择不属于 m 的所有索引:

    A[np.setdiff1d(np.arange(A.shape[0]), m)]
    

相关问题