机器学习实战读取数据

        
#读取数据模块。读取TXT文件，输出为测试集合、标签集
#1、读数据
#2、去回车，切割数据
#3、制作新的矩阵
#读取数据
def duqushuju(filename):
    f=open(filename)
    f1=f.readlines()
    index=0
    testdata=zeros((len(f1),3))
    label=[]
    for i in f1:
        yihang=i.strip()
        yihang1=yihang.split('\t')
        testdata[index,:]=yihang1[0:3]
        label.append(yihang1[-1])
    return label,testdata    
la,testd=duqushuju('datingTestSet2.txt')
print([int(i) for i in la])