三、Matplotlib——Legend 图例

 

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-3, 3, 50)
# print(x)
y1 = 2 * x + 1
y2 = x ** 2

plt.figure()

print("************设置坐标轴的范围、名字、下标值、文字描述、字体**************")
print("=========x 和y 的取值范围=========")
plt.xlim((-1, 2))
plt.ylim((-2, 3))
print("=========x 和y 的名称描述=========")
plt.xlabel("i am x")
plt.ylabel("i am y")
print("=========x 和y 的下标数值替换=========")
# x
new_x_ticks = np.linspace(-1, 2, 5)
print(new_x_ticks)
plt.xticks(new_x_ticks)
# y
# plt.yticks([-2, -1, 0, 1, 2, 3],["really bad$", "bad", "normal", "good", "really good"])
# 修改字体
# \alpha 转成数字符号的alpha
plt.yticks([-2, -1.8, -1, 1.22, 3],[r"$really\ bad$", 
            r"$bad$", r"$normal$", r"$good\ \alpha $", r"$really\ good$"])
print("=========================")
# 注意： 传入handles时，此处变量一定要加逗号
l1, = plt.plot(x, y1, label='down', color='red', linewidth=1.0, linestyle='--')
l2, = plt.plot(x, y2, label='up')
# handles 指定要添加图例的线
# labels 顺序与handles对应，指定图例线中的名字
# loc 指定图例在图中的位置，best 自己寻找最佳位置
plt.legend(handles=[l1, l2, ], labels=['aaa', 'bbb'], loc='best')

plt.show()