Python之numpy中怎么使用mask

Python之numpy中怎么使用mask?这个问题可能是我们日常学习或工作经常见到的。希望通过这个问题能让你收获颇深。下面是小编给大家带来的参考内容,让我们一起来看看吧!

创新互联是一家专业提供云冈企业网站建设,专注与成都网站制作、网站设计、H5响应式网站、小程序制作等业务。10年已为云冈众多企业、政府机构等服务。创新互联专业网站设计公司优惠进行中。

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')

Python之numpy中怎么使用mask

带条件选择替换,比如我们需要将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)

感谢各位的阅读!看完上述内容,你们对Python之numpy中怎么使用mask大概了解了吗?希望文章内容对大家有所帮助。如果想了解更多相关文章内容,欢迎关注创新互联行业资讯频道。


新闻名称:Python之numpy中怎么使用mask
路径分享:http://scyanting.com/article/gcjgce.html