2_2最近邻算法kNN(k_nearest_neighbor)——file2matrix_2_2

声明：《机器学习实战》代码详细注释和重构，以及相关的函数、模块和算法的解释；本文为博主原创文章，未经博主允许不得转载。

1.k-近邻算法（程序清单2-2)

*#代码：Peter Gong_shuai
*#中文注释：Gong_shuai
*#代码重构：Gong_shuai
*#函数注解：Gong_shuai
*#相关的函数、模块和算法的解释：Gong_shuai

2. 代码

#coding=utf-8#k-近邻算法（程序清单2-2）#源代码：Peter  Gong_shuai#中文注释：Gong_shuai#代码重构：Gong_shuai#函数注解：Gong_shuai#环境：python2.7from numpy import *import operatorfrom os import listdir #函数listdir可以列出给定目录的文件名#处理输入格式问题，输入为文件名字符串，输出为训练样本矩阵和类标签向量def file2matrix(filename):    fr = open(filename)    numberOfLines = len(fr.readlines())         #得到文本行数    returnMat = zeros((numberOfLines,3))        #创建以零填充的矩阵，为了简化，另外的一个维度设为3    classLabelVector = []                       #返回标签         fr = open(filename)    #解析文件数据到列表，循环处理文件中的每一行的数据    index = 0    for line in fr.readlines():        line = line.strip()    #截取掉所有的回车字符        listFromLine = line.split('\t')    #将整行数据分割成一个元素列表        returnMat[index,:] = listFromLine[0:3]     #选取前三个元素，存储到特征矩阵中        classLabelVector.append(int(listFromLine[-1]))        index += 1    return returnMat,classLabelVector# datingDataMat,datingLabels = file2matrix('datingTestSet.txt')# print datingDataMat