Python matplotlib画图实例之绘制拥有彩条的图表

2019-11-25 15:27:14

字体：大中小

来源：转载

供稿：网友

生产定制一个彩条标签。

首先导入：

import matplotlib.pyplot as pltimport numpy as npfrom matplotlib import cmfrom numpy.random import randn

制作拥有垂直（默认）彩条的图表：

fig, ax = plt.subplots()data = np.clip(randn(250, 250), -1, 1)cax = ax.imshow(data, interpolation='nearest', cmap=cm.coolwarm)ax.set_title('Gaussian noise with vertical colorbar')# Add colorbar, make sure to specify tick locations to match desired ticklabelscbar = fig.colorbar(cax, ticks=[-1, 0, 1])cbar.ax.set_yticklabels(['< -1', '0', '> 1']) # vertically oriented colorbar

效果图：

制作拥有水平彩条的图表：

fig, ax = plt.subplots()data = np.clip(randn(250, 250), -1, 1)cax = ax.imshow(data, interpolation='nearest', cmap=cm.afmhot)ax.set_title('Gaussian noise with horizontal colorbar')cbar = fig.colorbar(cax, ticks=[-1, 0, 1], orientation='horizontal')cbar.ax.set_xticklabels(['Low', 'Medium', 'High']) # horizontal colorbarplt.show()

效果图：