python處理txt文件，將文本數據轉化爲矩陣

fo = open('datingTestSet2.txt')
print ("文件名爲: ", fo.name)

filelist = fo.readlines()            #將txt文件轉換爲所有的行組成的列表  
numberoflines =len(filelist)         #得到行數            
print ("行數: %s" % (numberoflines))
returnMat = zeros((numberoflines,3)) #生成一個numberoflines行，3列的矩陣
print(returnMat)
classLabelVector =[]
index=0

returnMat[0,:] =[1,2,3]
returnMat[1,:] =[4,5,6]
for line in filelist:               #依次讀取每行
    line = line.strip()             #去掉每行頭尾空白 
    listline=line.split('\t')       #按換行符分割數據
            
    returnMat[index,:] =listline[0:3]  #將文本數據前三列存入數據矩陣
    classLabelVector.append(int(listline[3])) #第四列以整型存入標籤向量，append()函數用於向列表中添加元素
    #classLabelVector[index]=int(listline[3]) 錯誤，因爲沒有定義classLabelVector的長度
    index+=1
print(classLabelVector)
print(returnMat)
# 關閉文件
fo.close()

運行結果