控制pcolormesh的亮度

Question

我试图用python和matplotlib的pcolormesh绘制电子的概率（在氢原子中）。 一切都很顺利，因为分布下降得如此之快 - 一些细节是不可见的，例如，径向函数的零点（在较高能量状态下）的周围太褪色，使得很难注意到波函数实际上在某些半径消失。

我知道我可以通过一些重新调整和调整来解决这个问题。波函数，但我宁愿调整我的绘图技巧，并实现如何使用matplotlib。

我想调整热图，以便更多地图明亮。 有没有办法控制其灵敏度？

提前致谢。

Answer 1

你可以使用gamma correction做。我在非常相似的情况下使用它并获得了非常好的结果。

一种方法：

normalized = original/original.max()  # rescale to between 0 and 1
corrected = numpy.power(normalized, gamma) # try values between 0.5 and 2 as a start point
plt.imshow(corrected)

这是有效的，因为将0和1之间的间隔提升到给定的指数会产生单调递增的结果，即0,0和1,1。这类似于移动Photoshop / GIMP“级别”对话框的中间滑块。

编辑：更好的是，似乎Matplotlib已经有a class for that。