Python画散点图(Knn中数据)

来源:互联网 发布:手机能做淘宝链接 编辑:程序博客网 时间:2024/06/09 01:12

用之前做的KNN算法的数据画的散点图
我在这就只给出我变化之后main函数了
其实只是用了那个文件流的部分
可以在之前那个版本的knn算法中找到,或者你自己写一个就好了

if __name__ == "__main__":    group, labels = createDataSet()    x = [1, 1]    print(classify0(x, group, labels, 2))    group, labels = file2Matrix("文件地址")    x = [3000, 5, 0.9]    print(classify0(x, group, labels, 5))    datingDataMat = group    fig = plt.figure()    ax = fig.add_subplot(111)    ax.scatter(datingDataMat[:, 1], datingDataMat[:, 2])    plt.show()

主要部分是

    datingDataMat = group #拿到数据    fig = plt.figure()    # 画图专用    ax = fig.add_subplot(111)      ax.scatter(datingDataMat[:, 1], datingDataMat[:, 2])    plt.show()

add_subplot 参数的意思是
将整个figure分为1*1
然后输出下面那个东西在第一块中
scatter是画散点图的工具
左边是行坐标,右边是列坐标
最后一句一定要有,不然图出不来的
我不会告诉你,我经历了什么的