python plt可视化——打印特殊符号和制作图例

1、打印特殊符号

matplotlib在公式书写上面跟latex很相似，接下来我们就特殊符号，上标下标来具体展示一下。

import matplotlib.pyplot as plt

x = [i+1 for i in range(20)]
y = x
plt.figure()
plt.title(r'$\alpha$ > $\beta$')  # 打印α>β
plt.xlabel(r'$\theta$')  # 打印θ
plt.ylabel(r'$\omega$')  # 打印ω
plt.plot(x, y)
plt.show()

效果如下：

由此可见，打印特殊符号需要知道特殊符号的英文名称，在前面加上转义符反斜杠，再用一对美元符号包起来即可。

接下来，我们尝试打印下标和上标。下标需要加"_"，上标需要加"^"，这跟latex语法完全一样。

import matplotlib.pyplot as plt

x = [i+1 for i in range(20)]
y = x
plt.figure()
plt.title(r'$\alpha_i$ > $\beta_i$')  # 打印α_i > β_i
plt.xlabel(r'$\theta^2$')  # 打印θ^2
plt.ylabel(r'$\omega^n$')  # 打印ω^n
plt.plot(x, y)
plt.show()

我们看看效果：

更多符号对应字母请见下图：

2、制作图例，legend函数

import matplotlib.pyplot as plt
from math import sin, cos, exp

x = [(i+1)/100 for i in range(1000)]
y1 = [sin(i) for i in x]
y2 = [cos(i) for i in x]
y3 = [exp(-i) for i in x]

plt.figure()
plt.plot(x, y1)
plt.plot(x, y2)
plt.plot(x, y3)

# 分别对应y1，y2，y3标志图例，注意e^(-x)要加大括号表示(-x)是一个整体，都是上标
plt.legend(['sin(x)', 'cos(x)', '$e^{-x}$'])
plt.show()

在文章最后附上参考链接~
https://matplotlib.org/users/mathtext.html