Python,Numpy:如何选择具有给定掩码的Numpy数组
例如,给定一个(10000,250)大小的numpy矩阵aPython,Numpy:如何选择具有给定掩码的Numpy数组,python,arrays,numpy,set-difference,Python,Arrays,Numpy,Set Difference,例如,给定一个(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
>>>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])
您可以使用它来选择不属于m
的所有索引:
A[np.setdiff1d(np.arange(A.shape[0]), m)]