python如何导入txt文件-数据从txt文本导入python

机器学习实战 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])