kNN-画图

现在我们想要展示一些可视化内容

首先导包，如果是在jupyter notebook上，需要加入魔法函数：%matplotlib inline，这表示可以在jupyter上直接画图

import datingTest
import matplotlib.pyplot as plt
import numpy as np

这里导入的datingTest是一个datingTest.py文件，里面是程序清单2-2的内容

画图之前，我们需要获得我们的x和y，按书上2.2.2的内容，我们是要输出dataSet的第二列和第三列的内容

所以我们首先获得数据

dataSet = datingTest.dataSet
labelSet = datingTest.labelSet

然后进行画图并保存查看

# 获得画布
fig = plt.figure()
# 获得子图
ax = fig.add_subplot(111)
# 第二列和第三列以及标记，这里的15.0自己可以多修改几次看看输出的是什么
ax.scatter(dataSet[:,1],dataSet[:,2],15.0*np.array(labelSet),15.0*np.array(labelSet))
# 标签
plt.xlabel("Percentage of time spent playing video games")
plt.ylabel("The number of liters of ice cream consumed per week")
# 也可以直接使用plt，这里ax和plt直接画有一点区别，后面再详细学一下matplotlib
# plt.subplot(111)
# plt.scatter(dataSet[:,1],dataSet[:,2])
plt.show()
plt.savefig("./datingImg1.png")

其他问题：这里保存成png格式，打开的图片坐标系会是黑色的，虽然也可以改，但现在还不知道为什么

如果保存成jpg格式，则没有png格式的问题