如下所示：

# -*- coding: utf-8 -*-# @ author hulei 2016-5-3from numpy import *import operatorfrom os import listdir import sysreload(sys)sys.setdefaultencoding('utf8') # x,y=getDataSet_dz('iris.data.txt',4) def getDataSet(filename,numberOfFeature):  #將數據集讀入內存  fr = open(filename) numberOfLines = len(fr.readlines())   #get the number of lines in the file file.readlines()是把文件的全部內容讀到內存，并解析成一個list returnMat = zeros((numberOfLines,numberOfFeature))  #prepare matrix to return 3代表數據集中特征數目### classLabelVector = []      #prepare labels return  fr = open(filename) index = 0 for line in fr.readlines():  line = line.strip()     #strip() 參數為空時，默認刪除空白符（包括'/n', '/r', '/t', ' ')  listFromLine = line.split(',')   #split 以什么為標準分割一次 分成數組中的每個元素  returnMat[index,:] = listFromLine[0:numberOfFeature]   #classLabelVector.append(int(listFromLine[-1])) #append() 方法向列表的尾部添加一個新的元素  if listFromLine[-1] == 'Iris-setosa' :   classLabelVector.append(1)  elif listFromLine[-1] == 'Iris-versicolor' :   classLabelVector.append(2)  else:  #elif listFromLine[-1] == 'Iris-virginica' :   classLabelVector.append(3)  index += 1 return returnMat,classLabelVector def getDataSet_dz(filename,numberOfFeature): #改進版，可以消除數據中的空白行 numberOfLines = 0 mx = []  #將數據集 去除空行后存入 fr = open(filename) for line in fr.readlines():    line = line.strip()   if line != '' : #去除空白行    numberOfLines+=1   mx.append( line.split(',') ) returnMat = zeros((numberOfLines,numberOfFeature)) classLabelVector = []  for index in range(numberOfLines) :  returnMat[index,:] = mx[index][0:numberOfFeature]   if mx[index][-1] == 'Iris-setosa' :   classLabelVector.append(1)  elif mx[index][-1] == 'Iris-versicolor' :   classLabelVector.append(2)  else:  #elif listFromLine[-1] == 'Iris-virginica' :   classLabelVector.append(3) return returnMat,classLabelVector

以上這篇Python讀取數據集并消除數據中的空行方法就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持武林站長站。