Numpy中的mask的使用

numpy中矩阵选取子集或者以条件选取子集，用mask是一种很好的方法 简单来说就是用bool类型的indice矩阵去选择， mask = np.ones(X.shape[0], dtype=bool) X[mask].shape mask.shape mask[indices[0]] = False mask.shape X[mask].shape X[~mask].shape (678, 2) (678,) (678,) (675, 2) (3, 2) 例如我们这里用来选取全部点中KNN选取的点以及所有剩余的点 from sklearn.neighbors import Neare