在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()

我想要一条绿色（说）线出现在colourbar上的4.0处。有什么想法吗？

由于

Answer 1

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

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