matplotlib做图中常用的辅助函数

matplotlib可能是Python语言中最常用的绘图库了，使用它可以较为容易的做出印刷品质的专业图形，此外，matplotlib的定制程度也很高，可以满足各式各样的绘图要求，能够限制你做图能力只是你的想像力而已。然而，也正因为matplotlib的定制性，相关API较为底层，所以为了做出复杂的图形，常常需要写很多行代码，显然不够经济。为了节省画图的时间，我们可以把日常工作中经常碰到的某些绘图需求加以定制化抽象化，编写一些辅助函数，这些函数将某些绘图需求打包封装了。这样，下次我们绘图里只需要调用相关函数就可以很快的做出需要的图形了，少写很多代码，也更加符合DRY(Don’t Repeat Yourself)的原则。

在这篇文章里，我将对我在使用matplotlib绘图中编写的辅助函数做一个阶段性的总结，如果日后写了新的函数，我再进一步更新。

为了节省我们在绘图时的时间，我一般会在开始做图时，在jupyter notebook中做一些初始化设置，包括导入必须要的库，如numpy,matplotlib,seaborn等；设置图形的大小、风格等。你可以根据自己的需求对这部分配置进行修改，作为自己的初始化配置，下次绘图时可以直接使用。我的初始化配置如下：

import numpy as np
import matplotlib
import matplotlib.pyplot as plt
import warnings
warnings.filterwarnings("ignore")
%matplotlib inline
import pandas as pd
import seaborn as sns
sns.set(style="white",color_codes=True)
plt.rcParams['figure.figsize'] = (15,9.27)
# Set the font set of the latex code to computer modern
matplotlib.rcParams['mathtext.fontset'] = "cm"

一、绘图时使用latex

matplotlib支持在绘图时使用latex代码，这样便可以更好的支持在图形中加入数学公式。使用latex代码的方式也很简单，只需要在相应的字符串两边各加上一个$号，如$y=e^x$就可以显示为$y=e^x$。但是每次都要输入$号比较麻烦，我编写了一个辅助函数可以自动为字符串两端加上$，这样只需求调用函数，正常输入latex代码或者普通文本，就可以用latex公式的形式呈现了。如下：

def latex(s=''):
    n = len(s.split(' '))
    if n == 1:
        if s == '':
            return '$'+'\ '+'$'
        else:
            return '$'+s+'$'
    else:
        return '$'+'\ '.join(s.split(' '))+'$'