《python数据分析与挖掘实战》 ID3决策树

#-*- coding: utf-8 -*-
#使用ID3决策树算法预测销量高低
import pandas as pd
from sklearn.tree import DecisionTreeClassifier as DTC
from sklearn.tree import export_graphviz #导入相关函数，可视化决策树。
#from sklearn.externals.six import StringIO  书中导入的这个函数实际上没有用到

#参数初始化
filename = r'D:\python\stats\pythonsjwj\chapter5\demo\data\sales_data.xls'
data = pd.read_excel(filename, index_col = u'序号')  #导入数据

#数据是类别标签，要将它转换为数据
#用1来表示“好”“是”“高”这三个属性，用-1来表示“坏”“否”“低”
data[(data == u"