python_用布尔型索引来处理数组

用布尔型索引来处理数组

项目索引：
/media/HD12T/shoa/pycharm_space/tensorflow_NLP/algorithm/Boolean_index_test.py

图.左边图里面的点有10个，用布尔索引来处理数组后，返回7个点

from sklearn.datasets import make_blobs
import matplotlib.pyplot as plt

centers = [[1, 1], [-1, -1], [1, -1]]

X, labels_true = make_blobs(n_samples=10, centers=centers, cluster_std=0.7,random_state=170)
my_members = [True,False,True,True,True,False,False,True,True,True]
col = '#4EACC5'
print(X[:,0].ndim,X[:,0].shape)

fig = plt.figure(figsize=(8, 3))
fig.subplots_adjust(left=0.02, right=0.98, bottom=0.05, top=0.9)
ax = fig.add_subplot(1,2,1)
ax.plot(X[:,0],X[:,1],'w', markerfacecolor=col, marker='.' ,markersize=16)


ax = fig.add_subplot(1,2,2)
# ax.plot(X[:,0][my_members], X[my_members, 1], 'w', markerfacecolor=col, marker='.',markersize=16)
ax.plot(X[my_members,0], X[my_members, 1], 'w', markerfacecolor=col, marker='.',markersize=16)
#X[my_members,0] 等于X[:,0][my_members] ---> 布尔型索引
# plt.scatter(X[:,0],X[:,1])
plt.show()

更多参考资料：

python numpy 数组型索引 布尔型索引 2