控制亮度在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。