plt绘图

• 绘制散点图
  利用np.array类型的data前两列，根据lable分颜色绘制散点图

import matplotlab.pyplot as plt
plt.scatter(data[:,0],data[:,1],c=15*label)
plt.show

效果图：

如果想画的精细一点儿，画出对应颜色的图例，则可以利用下面的代码

l=len(label)
for i in range(l):
    if label[i]==3:
        s1=plt.scatter(data[i, 0], data[i, 1], color='r',marker='o')
    elif label[i]==2:
        s2=plt.scatter(data[i, 0], data[i, 1], color='g',marker="*")
    else:
        s3=plt.scatter(data[i, 0], data[i, 1], color='y',marker="^")
plt.legend((s1,s2,s3),('didntLike','largeDoses','smallDoses'),loc='best')
plt.xlabel('X')
plt.ylabel('Y')
plt.show()

其中plt.scatter的参数market设置点形状，color设置点颜色。plt.legend设置图例

效果图：