python农产品价格预测_【Machine Learning】决策树案例：基于python的商品购买能力预测系统...

fromsklearn.feature_extraction import DictVectorizer

import csvfromsklearn import treefromsklearn import preprocessingfrom sklearn.externals.six import StringIO

读取csv文件，将其特征值存储在列表featureList中，将预测的目标值存储在labelList中

'''

Description:python调用机器学习库scikit-learn的决策树算法，实现商品购买力的预测，并转化为pdf图像显示

Author:Bai Ningchao

DateTime:2016年12月24日14:08:11

Blog URL:http://www.cnblogs.com/baiboy/

'''

def trainDicisionTree(csvfileurl):

'读取csv文件，将其特征值存储在列表featureList中，将预测的目标值存储在labelList中'

featureList = []

labelList = []

#读取商品信息

allElectronicsData=open(csvfileurl)

reader = csv.reader(allElectronicsData) #逐行读取信息

headers=str(allElectronicsData.readline()).split(',') #读取信息头文件

print(headers)

运行结果：