matplotlib: colorbars和其文本标签

23 浏览
0 Comments

matplotlib: colorbars和其文本标签

我想为一个热图创建一个带有颜色标注的图例，使得每个离散颜色的标签位于中心位置。从这里借用了一个示例：

import matplotlib.pyplot as plt
import numpy as np
from matplotlib.colors import ListedColormap
#离散的颜色方案
cMap = ListedColormap(['white', 'green', 'blue','red'])
#数据
np.random.seed(42)
data = np.random.rand(4, 4)
fig, ax = plt.subplots()
heatmap = ax.pcolor(data, cmap=cMap)
#图例
cbar = plt.colorbar(heatmap)
cbar.ax.set_yticklabels(['0','1','2','>3'])
cbar.set_label('联系次数', rotation=270)
#将主要刻度放在每个单元格的中间
ax.set_xticks(np.arange(data.shape[1]) + 0.5, minor=False)
ax.set_yticks(np.arange(data.shape[0]) + 0.5, minor=False)
ax.invert_yaxis()
#标签
column_labels = list('ABCD')
row_labels = list('WXYZ')
ax.set_xticklabels(column_labels, minor=False)
ax.set_yticklabels(row_labels, minor=False)
plt.show()

这将生成以下图表：

pmesh plot

理想情况下，我想生成一个带有四种颜色的图例条，并且对于每种颜色，在其中心有一个标签：0,1,2,>3。如何实现这个目标？

0