欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  web前端

Python-Passing a matplotlib figure to HTML (flask)_html/css_WEB-ITnose

程序员文章站 2022-06-09 23:05:47
...
最近因工作需要,需要将决策树生成的图在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 = '''

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是错的,哪个叶子结果是错的等功能。