【Python】matplotlib绘制图像时增加颜色条

一、需求

plt.imshow()是matplotlib中的一个函数，用于显示图像。它可以传递一个二维或三维数组作为image参数， 并将图像数据显示为图形，并对图像进行不同的可视化设置。

在显示的过程中，我们如果需要增加一个图例显示颜色条，可以使用自带的colorbar实现。

代码如下：

  fig, ax = plt.subplots()
  im = ax.imshow(pm2_5, cmap="coolwarm", vmax=255, vmin=0)
  fig.colorbar(im ,ax=ax)

二、实现效果