任意颜色条

时间:2016-07-01 13:32:15

标签: python matplotlib colorbar

我的数据在-70,0范围内,我使用imshow()显示,并希望使用非线性颜色条来表示数据,因为我在-70,-60范围内都有paterns和-70,0范围。 我想用最简单的方法使用任意函数(参见示例)对colorbar进行重新缩放/重新规范化,以便所有的paterns都能很好地显示。

以下是数据和功能的示例:

sample_data=(np.ones((20,20))*np.linspace(0,1,20)**3)*70-70

def renorm(value):
    """
    Example of the way I would like to adjust the colorbar but it might as well be an arbitrary function
    Returns a number between 0 and 1 that would correspond to the color wanted on the original colorbar
    For the cmap 'inferno' 0 would be the dark purple, 0.5 the purplish orange and 1 the light yellow
    """

    return np.log(value+70+1)/np.log(70+1)

expected colorbar

1 个答案:

答案 0 :(得分:1)

这就是我设法做的事情:

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.colors import PowerNorm
sample_data=(np.ones((20,20))*np.linspace(0,1,20)**3)*70-70
plt.figure()
im = plt.imshow(sample_data+70, norm=PowerNorm(gamma=0.5))
cbar = plt.colorbar(orientation='horizontal')
cbar.ax.set_xticklabels(np.arange(-70, 0, 8))
plt.show()

enter image description here

您可以更改gamma。 但是,不建议使用此类可视化,请参阅:http://matplotlib.org/users/colormapnorms.html 在“权力法则”下 - > “注意”