深圳罗湖建网站,搜狗网站制作,网店设计英文,免费申请网号假设data是m行两列的训练样本#xff0c;labels是m行一列的类标签#xff0c;类标签一共有3类#xff0c;分别用1、2、3表示#xff0c;现将data用散点图表示出来#xff0c;且不同类的样本有不同的颜色#xff1a;
import matplotlib.pyplot as pltfig plt.figure()
a…假设data是m行两列的训练样本labels是m行一列的类标签类标签一共有3类分别用1、2、3表示现将data用散点图表示出来且不同类的样本有不同的颜色
import matplotlib.pyplot as pltfig plt.figure()
ax fig.add_subplot(111) # 创建一个一行一列的图
ax.scatter(data[:, 0], data[:, 1], 15.0*np.array(labels), 15.0*np.array(labels)) # 15.0是散点的大小
plt.show() 为了得到更好的效果并以红色的*表示类标签1、蓝色的o表示表示类标签2、绿色的表示类标签3修改参数如下
import numpy as np
import matplotlib.pyplot as pltfig plt.figure()
ax fig.add_subplot(111)
labels np.array(labels)
idx_1 np.where(labels 1) # 找出第一类
p1 ax.scatter(data[idx_1, 0], data[idx_1, 1], marker*, colorr, label1,s20)
idx_2 np.where(labels 2) # 找出第二类
p2 ax.scatter(data[idx_2, 0], data[idx_2, 1], marker o,color b,label2,s10)
idx_3 np.where(labels 3) # 找出第三类
p3 ax.scatter(data[idx_3, 0], data[idx_3, 1], marker ,color g,label3,s30)
plt.legend(locupper right)
plt.show()
