Numpy中的mask
numpy中矩陣選取子集或者以條件替換,用mask是一種很好的方法 簡單來說就是用bool類型的indice矩陣去選擇, 1 2 3 4 5 6 7 8 9 10 11 12 mask = np.ones(X.shape[ X[mask].shape mask.shape mask[indices[ mask.shape X[mask].shape X[~mask].shape ( ( ( ( ( 例如我們這裡用來選取全部點中KNN選取的點以及所有剩餘的點 1 2 3 4 5 6 7 from nbrs = NearestNeighbors( _,indices = nbrs.kneighbors(X) mask = np.ones(X.shape[ mask[indices[ plt.scatter(X[mask][:, plt.scatter(X[~mask][:, 帶條件選擇替換,比如我們需要將a矩陣內某條件的行置換為888剩餘置換為999,可以直接用mask或者再用where一步搞定: 1 2 3 4 5 …