当用sklearn库来实现决策树算法时,
from sklearn.feature_extraction import DictVectorizer import csv from sklearn import preprocessing from sklearn import tree from sklearn.externals.six import StringIO allelectronicsData = open(r"E:\MaiziEdu\AllElectronics.csv",'r') reader = csv.reader(allelectronicsData) headers = next(reader) print(headers)
在Python 2.x版本时是用 headers = reader.next();
而3.x版本,则需要使用headers = next(reader)