Python学习网 发表于 2020-11-3 15:38:16

Python常见问题:numpy中mask的基础问题

有刚入门的小白不知道numpy中如何使用mask,今天小编就来讲讲使用mask会遇到的一些问题。
numpy中矩阵选取子集或者以条件选取子集,用mask是一种很好的方法。
简单来说就是用bool类型的indice矩阵去选择。
<p style="line-height: 1.75em;"><span style="font-family: 微软雅黑, "Microsoft YaHei";">mask = np.ones(X.shape, dtype=bool)

X.shape

mask.shape

mask] = False

mask.shape

X.shape

X[~mask].shape

(678, 2)

(678,)

(678,)

(675, 2)

(3, 2)<br></span></p>

例如我们这里用来选取全部点中KNN选取的点以及所有剩余的点。
<p style="line-height: 1.75em;"><span style="font-family: 微软雅黑, "Microsoft YaHei";">from sklearn.neighbors import NearestNeighbors

nbrs = NearestNeighbors(10).fit(X)

_,indices = nbrs.kneighbors(X)

mask = np.ones(X.shape, dtype=bool)

mask] = False

plt.scatter(X[:,0],X[:,1],c='g')

plt.scatter(X[~mask][:,0],X[~mask][:,1],c='r')<br></span></p>

带条件选择替换,比如我们需要将a矩阵内某条件的行置换为888剩余置换为999,可以直接用mask或者再用where一步搞定:
<p style="line-height: 1.75em;"><span style="font-family: 微软雅黑, "Microsoft YaHei";">mask = np.ones(a.shape,dtype=bool) #np.ones_like(a,dtype=bool)

mask = False

a[~mask] = 999

a = 888

#############

np.where(mask, 888, 999)<br></span></p>
是不是很容易呢,小伙伴们学会了没有~更多Python学习推荐:PyThon学习网教学中心。
页: [1]
查看完整版本: Python常见问题:numpy中mask的基础问题