机器学习实战 2txt文件读取数据（书2.2.1）

最新推荐文章于 2024-05-28 17:03:46 发布

原创

最新推荐文章于 2024-05-28 17:03:46 发布 · 1.4k 阅读

2 ·

CC 4.0 BY-SA版权

本文档主要介绍了在机器学习实战中遇到的2txt文件读取问题，包括'reload()'函数的使用错误及datingTestSet2.txt文件的使用。在Python环境中，不再需要'reload()'函数，直接导入即可。同时指出了书中datingTestSet.txt文件名的错误，应更正为datingTestSet2.txt。

书2.2.1 解析txt文件读取其中的数据

1.记录

def file2matrix(filename):
    fr = open(filename)
    arrayOLines = fr.readlines()
    numberOfLines = len(arrayOLines)
    returnMat = zeros((numberOfLines,3))
    classLabelVector = []
    index = 0
    for line in arrayOLines:
        line = line.strip()
        listFromLine = line.split('\t')
        returnMat[index,:] = listFromLine[0:3]
        classLabelVector.append(int(listFromLine[-1]))
        index += 1
    return returnMat,classLabelVector