Python 将每行的前len-1列数据抽成一个矩阵，将最后的一列作为另一个矩阵

文件解析与数据组织

原创于 2017-09-03 11:53:02 发布 · 2.5k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#python #机器学习

python 同时被 3 个专栏收录

37 篇文章

订阅专栏

机器学习

19 篇文章

订阅专栏

python常用写法

2 篇文章

订阅专栏

本文介绍了一个Python函数loadDataSet(fileName)，该函数从指定文件中读取tab分隔的数据，并将其组织为两个矩阵：一个包含所有行的前len-1列数据，另一个包含最后一列数据。示例展示了如何使用此函数处理包含两列数据和一列标签的文件。

作用：将每行的前len-1列数据抽成一个矩阵，将最后的一列作为另一个矩阵

每行各数据间隔是tab(\t
)

def loadDataSet(fileName):      #general function to parse tab -delimited floats
    numFeat = len(open(fileName).readline().split('\t')) - 1 #get number of fields
    print "numFeat:",numFeat
    dataMat = []; labelMat = []
    fr = open(fileName)
    for line in fr.readlines():
        lineArr =[]
        curLine = line.strip().split('\t')
        for i in range(numFeat):
            print "i:",i
            lineArr.append(float(curLine[i]))
        dataMat.append(lineArr)
        labelMat.append(float(curLine[-1]))
    return dataMat,labelMat

数据：

1.000000	0.067732	3.176513
1.000000	0.427810	3.816464

输出结果：

矩阵1：

dataMat:[[1.0, 0.067732], [1.0, 0.42781]]

矩阵2：

labelMat: [3.176513, 3.816464]