python - 控制 pcolormesh 中的亮度

Question

我正在尝试使用 python 和 matplotlib 的 pcolormesh 绘制电子的概率(在氢原子中)。一切都很好，除了由于分布下降得如此之快 - 一些细节不可见，例如，径向函数的零点周围(在较高能态)太淡化，使得很难注意到波函数实际上在某些半径处消失。

我知道我可以通过对波函数进行一些重新缩放和“调整”来处理这个问题，但我宁愿调整我的绘图技能并了解如何使用 matplotlib 来做到这一点。

我想调整热图，使 map 的更多部分变得明亮。有没有办法控制它的灵敏度？

提前致谢。

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“级别”对话框的中间 slider 。

编辑:更好的是，Matplotlib 似乎已经有了 a class for that .