将文本转化为Numpy的矩阵

最新推荐文章于 2022-07-26 11:52:00 发布

原创最新推荐文章于 2022-07-26 11:52:00 发布 · 801 阅读

1 ·

CC 4.0 BY-SA版权

本文介绍了一种将文本文件内容转换为Numpy矩阵的方法。通过读取文件的每一行，并将其拆分为元素，最终形成一个二维矩阵。此方法适用于处理包含表格数据的文本文件。

将文本转化为Numpy的矩阵

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('\t')
        returnMat[index,:] = listFromLine[0:3]
        classLabelVector.append(int(listFromLine[-1]))
        index += 1
    return returnMat,classLabelVector