在练习机器学习的时候遇到了一个小问题,花费了挺长时间,故记录下
源码
from numpy import *
def file2matrix(filename):
fr = open(filename)
arrayOLines = fr.readlines()
numberOFLines = len(arrayOLines)
returnMat = zeros(numberOFLines)
classLabelVector = []
index = 0
for line in arrayOLines:
line = line.strip()
listFromLine = line.split('\t')
returnMat[index,:] = listFromLine[0:3]
classLabelVector.append(int(listFromLine[-