图表辅助元素定制

 实例：绘制一个包含正弦曲线和余弦曲线的图表。

需求：（1）正弦曲线的样式：红色、线宽为1.0；
           （2）余弦曲线的样式：蓝色、线宽为1.0，透明度为0.5；
           （3） x轴的刻度标签为“-π”、 “-π/2”、 “0”、 “π/2”、 “-π”；
           （4） 在x=1、y=np.cos(1)的位置添加指向型注释文本；
           （5）  为正弦曲线和余弦曲线封闭起来的区域填充绿色，透明度为0.25。

 一、导入模块并准备数据

#导入模块
import numpy as np
import matplotlib.pyplot as plt
#准备数据
x = np.linspace(-np.pi, np.pi, 256, endpoint=True)  
y_cos, y_sin = np.cos(x), np.sin(x)

二、创建画布 

fig = plt.figure()
ax = fig.add_subplot(111)

三、绘制折线图 

plt.plot(x,y_cos,linewidth = 1.0,color='blue',alpha=0.5)
plt.plot(x,y_sin,linewidth = 1.0,color='red')

四、添加辅助元素

#添加图表标题
ax.set_title("Cos&Sin")
#设置标签刻度
ax.set_xticks([-np.pi, -np.pi/2, 0, np.pi/2, np.pi],
              [r'$-\pi$', r'$-\pi/2$', r'$0$', r'$\pi/2$', r'$\pi$'])
ax.set_yticks([1.0, 0.5, 0.0, -0.5, -1.0])
#添加图例
ax.legend(loc ='upper left')
#添加网格
ax.grid(linewidth=0.5)
#在x=1、y=np.cos(1)的位置添加指向型注释文本
plt.annotate('cos(1)',(1,np.cos(1)),(1.35,np.cos(1)+0.25),
             arrowprops={'arrowstyle':'->'})

 五、图表样式美化

ax.fill_between(x,y_cos,y_sin,y_sin<y_cos,color = 'green',alpha =0.25)

代码运行结果展示 

 

附加内容：颜色映射表分类