在matplotlib中,imshow方法用于绘制热图,基本用法如下
import matplotlib.pyplot as plt
import numpy as np
np.random.seed(123456789)
data = np.random.rand(25).reshape(5, 5)
plt.imshow(data)
输出结果如下
imshow方法首先将二维数组的值标准化为0到1之间的值,然后根据指定的渐变色依次赋予每个单元格对应的颜色,就形成了热图。对于热图而言,通常我们还需要画出对应的图例,图例通过colorbar方法来实现,代码如下
plt.imshow(data)
plt.colorbar()
输出结果如下
imshow方法常用的几个参数如下
1. cmap
cmap是colormap的简称,用于指定渐变色,默认的值为viridis, 在matplotlib中,内置了一系列的渐变色,用法如下
plt.imshow(data, cmap='Greens')
输出结果如下
完整的内置colormap的列表见如下链接
https://matplotlib.org/tutorials/colors/colormaps.html
2. aspect
aspect用于指定热图的单元格的大小&#x