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 NearestNeighbors
nbrs = NearestNeighbors(10).fit(X)
_,indices = nbrs.kneighbors(X)
mask = np.ones(X.shape[0], dtype=bool)
mask[indices[0]] = False
plt.scatter(X[mask][:,0],X[mask][:,1],c='g')
plt.scatter(X[~mask][:,0],X[~mask][:,1],c='r')
带条件选择替换，比如我们需要将a矩阵内某条件的行置换为888剩余置换为999，可以直接用mask或者再用where一步搞
定：
mask = np.ones(a.shape,dtype=bool) #np.ones_like(a,dtype=bool)
mask[indices] = False
a[~mask] = 999
a[mask] = 888
#############
np.where(mask, 888, 999)
您可能感兴趣的文章您可能感兴趣的文章:NumPy 数学函数及代数运算的实现代码Python实现曲线拟合操作示例【基于numpy,scipy,matplotlib
库】浅谈利用numpy对矩阵进行归一化处理的方法numpy.std() 计算矩阵标准差的方法对numpy数据写入文件的方法讲解
NumPy.npy与pandas DataFrame的实例讲解numpy的文件存储.npy .npz 文件详解使用Numpy读取CSV文件,并进行行列删除
的操作方法python 借助numpy保存数据为csv格式的实现方法
资源评论
weixin_38750007
- 粉丝: 4
- 资源: 921
