最近在学K近邻算法,离散点的绘制,代码 如下:
import numpy as np
import matplotlib.pyplot as plt
def createDateSet():
group =np.array([[1.0,2.0],[1.2,0.1],[0.1,1.4],[0.3,3.5],[1.1,1.0],[0.5,1.5]])
lables=np.array(['A','A','B','B','A','B'])
return group,lables
if __name__=='__main__':
group,lables =createDateSet()
plt.scatter(group[lables=='A',0],group[lables=='A',1],color='r',marker='*')
plt.scatter(group[lables == 'B', 0], group[lables == 'B', 1], color='g', marker='o')
plt.show()
刚开始学习,Python也不是很懂,其中
plt.scatter(group[lables=='A',0],group[lables=='A',1],color='r',marker='*')
```这部分不知怎么理解,不知X,Y值是如何与标签对应起来的特别是这句,
group[lables=='A',0],group[lables=='A',1]