【matplotlib】可视化解决方案——如何使用数学公式

数学公式

一.前言

当进行可视化的时候，经常会遇到要绘制一些公式的情况，这些公式长出现在图例、轴标签和标题中，问题在于书写公式如果使用普通的字符串显示效果十分的差，对比如下：

可以看到左图的图例不正规也不美观，而是用了LaTeX的公式显示效果就有了大幅度的提升。

二.解决方案

Matplotlib 的内部 LaTeX 解析器和布局引擎。所以可以直接使用LaTeX进行公式书写。

画两个函数为例：

• ${\sum }_1^{n}x+3$
• $\sin x\cos x$

完整程序如下：

import matplotlib.pyplot as plt

import numpy as np

plt.rcParams['font.sans-serif'] = ['SimHei']  # 设置支持中文
plt.rcParams['axes.unicode_minus'] = False  # 设置-号

# 获取X和Y
n = 15
x_1 = np.arange(1, n+1, 1)
y_1 = np.cumsum(x_1) + 3

x_2 = np.arange(0, 2*np.pi, np.pi/36)
y_2 = np.sin(x_2) * np.cos(x_2)

figure, ax = plt.subplots(1, 2)
# 绘图
ax[0].plot(x_1, y_1, lw=3, color='k', label='$\sum_{1}^{n}x+3}$')
ax[0].set_title('$\sum_{1}^{n}x+3}$函数', fontsize=20)
ax[0].set_xlabel("X", fontsize=15)
ax[0].set_ylabel('$\sum_{1}^{n}x+3}$函数', fontsize=15)
ax[0].legend()

ax[1].plot(x_2, y_2, lw=3, color='r', label='$\sin x \cos x$')
ax[1].set_title('$\sin x \cos x$函数', fontsize=20)
ax[1].set_xlabel("X", fontsize=15)
ax[1].set_ylabel('$\sin x \cos x$函数', fontsize=15)
ax[1].legend()

plt.show()

2.1图例使用数学公式

2.2标题使用数学公式

2.3轴标签使用数学公式

三.参考

1. 柱体标注问题解决方案
2. linestyle参数详解
3. capstyle参数详解
4. joinstyle参数详解
5. 子图间距问题解决方案
6. 子图设置大标题解决方案
7. 画图刻度设置解决方案