最近因工作需要，需要将决策树生成的图在html上展示，给专业的人员用来做修改。有几个坑，现在记录下。
再尝试了R的rpart，scikit-learn的decision tree之后，发现效果都跟业务要求有差距，因此决定自己用python3写决策树算法，因而很自然的用了matplotlib来做决策树的图像化。同时，flask是一种轻量级的web应用，很简便。所以最后问题就是怎么把matplotlib生成的图上传到页面上。
1. 安装matplotlib
在本机mac上安装matplotlib很方便，pip就行，但是在服务器linux上安装时，提示我linux上缺少freetype，libpng，在研究了半天之后，找到了解决方法。
安装freetype
wget http://download.savannah.gnu.org/releases/freetype/freetype-2.4.10.tar.gz
tar zxvf freetype-2.4.10.tar.gz
cd freetype-2.4.10/
./configure
make
sudo make install
安装libpng：
sudo yum install libpng-devel
在安装完上述包之后，再用pip install就好。
2.基本框架
需要的包：
import matplotlib
matplotlib.use('Agg') #不出现画图的框
from flask import Flask
app = Flask(__name__)
from io import BytesIO
import os
import base64
画图的函数：
def plotTree(s):
fig = plt.figure(1,figsize=(12,6),facecolor='white')
....#一系列画图操作
return fig
flask的函数:
@app.route('/', methods=['GET']) #传入参数
def index(name):
fig=plotTrees(name)
# Encode image to png in base64
sio = BytesIO()
fig.savefig(sio, format='png')
data=base64.encodebytes(sio.getvalue()).decode()
return html.format(data)
main函数:
if __name__ == '__main__':
html = '''
<html>
<body>
<img src="data:image/png;base64,{}" /></body>
<html>
app.run(port=7000,debug=True,threaded=False)
3 matplotlib的中文显示问题
在上述两步完成以后，其实已经在网页上显示图片了，但是由于matplotlib默认字体不支持中文，所以的中文字都变成了框框。有两种方法解决：
1. mac上成功解决的方法：
from pylab import *
mpl.rcParams['font.sans-serif'] = ['SimHei'] #指定默认字体
mpl.rcParams['axes.unicode_minus'] = False #解决保存图像是负号'-'显示为方块的问题
2. 该方法我在linux上没成功，研究半天没搞明白为啥，不过还好有另外的方法
1）首先在linux找系统的中文字体的.ttc文件
fc-list
如果不幸发现linux上没有中文字体，没事，将自己电脑上的字体复制到linux上就行，linux上字体放在/usr/share/fonts上。
2）然后在python中加入代码，表示matplotlib用该字体，比如我找的是wqy-zenhei/wqy-zenhei.ttc， 那么在python中加入
myfont = FontProperties(fname='/usr/share/fonts/wqy-zenhei/wqy-zenhei.ttc’)
然后在所有画图中添加字体的函数中，加入 fontproperties=myfont，然后就ok拉。
最后展示下很简单的结果：
后续有时间的话，会加入互动功能，让检查人员能反馈给我，哪条路径是错的，哪个node是错的，哪个叶子结果是错的等功能。
网友评论