机器学习实战 p21
源代码:
def file2matrix(filename):
fr = open(filename)
numberOfLines = len(fr.readlines()) #get the number of lines in the file
returnMat = zeros((numberOfLines,3)) #prepare matrix to return
classLabelVector = [] #prepare labels return
fr = open(filename)
index = 0
for line in fr.readlines():
line = line.strip()
listFromLine = line.split(' ')
returnMat[index,:] = listFromLine[0:3]
classLabelVector.append(int(listFromLine[-1])) 此句报错
index += 1
return returnMat,classLabelVector
报错如下:
>>> mat,label = kNN.file2matrix('datingTestSet.txt')
Traceback (most recent call last):
File "", line 1, in
File "kNN.py", line 50, in file2matrix
classLabelVector.append(int(listFromLine[-1])