python - 在 matplotlib colorbar 上指示外部值

Question

我有一个关于第三个变量的彩色散点图。我可以为这个图创建一个颜色条，但也想在颜色条上指示一个外部级别(只用一条水平线)。

X = np.array([0, 1, 2, 3, 4, 5, 6])
Y = np.array([0, 1, 2, 3, 4, 5, 6])
Z = np.array([0, 1, 2, 3, 4, 5, 6])
fig, ax = plt.subplots()
plt.scatter(X, Y, c=Z, cmap='gist_heat', vmin=Z.min(), vmax=Z.max(), s=500)
cbar = plt.colorbar()

我希望颜色条上的 4.0 处出现一条绿色(比方说)线。有什么想法吗？

谢谢

Answer 1

颜色条是一个轴。您可以在其上绘制一条水平线。要找出绘制线条的位置，请使用颜色条的范数将您的“特殊”值映射到它在颜色条轴上的位置:

cbar.ax.hlines(cbar.norm(4), 0, 1, color='green')