用面向对象思路为python数据可视化的图表添加辅助元素的定制

import matplotlib.pyplot as plt
import numpy as np
# 显示中文
plt.rcParams['font.sans-serif']=['SimHei']
plt.rcParams['axes.unicode_minus']=False
# 创建画布
f=plt.figure()


# 在画布上添加绘图区域
a=f.add_subplot(111)


# 准备数据

x = np.linspace(-np.pi, np.pi, 256, endpoint=True)
y1, y2 = np.sin(x), np.cos(x)
legend_labels = ['正弦曲线','余弦曲线']
# 绘制图表
lines = plt.plot(x, y1, x, y2)
# 添加图表辅助元素的定制
# 设置坐标轴标签
a.set_xlabel('x轴')
a.set_ylabel('x轴')
# 设置x轴的刻度范围和刻度标签
a.set_xlim(x.min() * 1.5, x.max() * 1.5)        # 刻度范围
a.set_xticks([-np.pi, -np.pi / 2, 0,  np.pi / 2, np.pi])       # 刻度位置
a.set_xticklabels([r'$-\pi$', r'$-\pi/2$', r'$0$', r'$\pi/2$', r'$\pi$'])      # 刻度标签
# 添加标题
a.set_title('正弦曲线和余弦曲线')
# 添加图例
a.legend(lines,legend_labels,shadow=True,fancybox=True)
# 显示水平网格
a.grid(axis='y',lw=0.3)
# 添加参考线
a.axhline(y=0,linestyle='--')     # 水平参考线
a.axvline(x=0,linestyle='--')     # 垂直参考线
# 添加参考区域
a.axhspan(ymin=0.5,ymax=1,alpha=0.4)       # 水平参考区域
a.axvspan(xmin=0.5,xmax=2,alpha=0.4)       # 垂直参考区域
# 添加注释文本
# 指向型注释文本
a.annotate('最小值',(-np.pi/2,-1),(-np.pi/2,-0.5),arrowprops={'arrowstyle':'->'})
# 添加无指向型注释文本
a.text(np.pi, 0.25, 'y=sin(x)',bbox={'alpha':0.25})
# 添加自定义样式的表格
a.table(cellText=[[-1,1],[-1,1]],rowLabels=['最小值','最大值'],colLabels=['正弦','余弦'],
           colWidths=[0.1]*3,loc='lower right')
# # 展示图表
plt.show()