学习Machine_learning系列

当用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)

猜你喜欢

转载自blog.csdn.net/dengdengma520/article/details/79591876