python - 使 plt.colorbar 扩展到 vmin/vmax 前后的步骤

Question

我想用 plt.hist2d 和 plt.colorbar 做点什么，但我真的很难弄清楚如何去做。为了解释，我编写了以下示例:

import numpy as np
from matplotlib import pyplot as plt

x = np.random.random(1e6)
y = np.random.random(1e6)

plt.hist2d(x, y)
plt.colorbar()

plt.show()

此代码生成的绘图类似于下图。

如果我生成直方图，理想情况下我希望颜色条超出数据的最大和最小范围延伸到超出最大和最小值的下一步。在此问题的示例中，这会将颜色条范围设置为从 9660 到 10260，增量为 60。

如何强制 plt.hist2d 或 plt.colorbar 设置颜色条，以便将刻度分配给绘制的颜色条的开始和结束？

Answer 1

我想这就是您要找的:

h = plt.hist2d(x, y)
mn, mx = h[-1].get_clim()
mn = 60 * np.floor(mn / 60.)
mx = 60 * np.ceil(mx / 60.)
h[-1].set_clim(mn, mx)
cbar = plt.colorbar(h[-1], ticks=np.arange(mn, mx + 1, 60), )

这给出了类似的东西，

使用 matplotlib.ticker 中的代码和代码的 tick_values 方法通常也很方便，但出于此目的，我认为上面的代码最方便。

祝你好运!