使用颜色映射表
matplotlib 主要有两种使用颜色映射表的方式:第一种方式是在调用函数或方法绘制图表或添加辅助元素时将颜色映射表传递给关键字参数cmap;第二种方式是直接调用set_cmap0函数进行设置。这两种方式的具体用法如下。
(1)使用关键字参数cmap的示例代码如下:
plt.scatter (x,y,c-np.random.rand(10), cmap-matplotlib.cm.jet)
(2)使用set_cmap0函数的示例代码如下:
plt.set cmap(matplotlib,cm.jet)
颜色映射表分类如图:
需求:绘制一个包含正弦曲线和余弦曲线的图表,具体要求如下:
(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
# 1.准备数据
x = np.linspace(-np.pi, np.pi, 256, endpoint=True)
y_cos, y_sin = np.cos(x), np.sin(x)
# 2.绘制折线
plt.plot(x, y_cos, color='blue', linewidth=1.0, alpha=0.5)
plt.plot(x, y_sin, color='red', linewidth=1.0)
# 3.图表辅助元素定制(图表标题,刻度标签,图例,网格,指向型注释文本)
# 添加图表标题
plt.title('Cos & Sin', loc='center')
# x轴的刻度标签
plt.xticks([-np.pi, -np.pi/2, 0, np.pi/2, np.pi],
[r'$-\pi$', r'$-\pi/2$',
r'$0$',
r'$\pi/2$', r'$\pi$']) #编写数学表达式参考matplotlib的官网)
# y轴的刻度标签
plt.yticks([-1.0, -0.5, 0.0, 0.5, 1.0])
# 添加图例
lines = plt.plot(x, y_cos, x, y_sin)
plt.legend(lines, ['COS', 'SIN'], ncol=1, loc='upper left')
# 添加网格
plt.grid( )
# 添加指向型注释文本
plt.annotate("Cos(1)",
xy=(np.pi/3, 0.55),
xytext=((np.pi/2.6) ,0.8),
arrowprops={'arrowstyle':'->'})
# 4.图表样式美化(填充颜色)
plt.fill_between(x, y_cos, y_sin, y_cos > y_sin, color='g', alpha=0.25)
plt.show()
运行效果如图: