如何用决策树生成树状图

import pandas as pd
from sklearn.tree import DecisionTreeClassifier as DTC
from sklearn.tree import export_graphviz

2.导入数据

filename='/data/python13/sales_data.txt'
data=pd.read_csv(filename,index_col='序号')

3.数据预处理

data[data=='好']=1
data[data=='是']=1
data[data=='高']=1
data[data!=1]=-1

4.特征提取

data[data=='好']=1
data[data=='是']=1
data[data=='高']=1
data[data!=1]=-1

5.建立决策树模型

dtc=DTC(criterion="gini").fit(x,y)

6.模型可视化

with open( 'tree.dot','w') as f:
    f=export_graphviz(dtc,feature_names=data.columns,out_file=f)

7.在linux中安装graphviz

sudo apt-get install graphviz

8.生成png图片

dot -Tpng tree.dot -o tree.png

9.在pycharm中打开该图片