python - 在颜色条附近绘制参考？

Question

我想在颜色条的右侧绘制一个正方形作为具有相同颜色编码的参考（见下图）。

但我找不到实现这个目标的方法。有没有善良聪明的人可以做到这一点？

标签： pythonmatplotlibheatmap

Answer 1

如果有人可能有与我相同的问题，请将其发布在这里。为了从颜色栏中获取颜色，我计算了颜色栏中给定参考的相应比例。
cmap = cm.get_cmap("OrRd") # get the corresponding colorbar reference = 90 # set the reference rgb = cmap( (reference - vmin) / (vmax - vmin) ) # find the color in the colorbar

最后，将其设置为“SquareObjectHandler”类中的“facecolor”。位置将是相同的方式。找出颜色条的坐标并相应地在“图例”中设置“bbox_to_anchor”。