Search code examples
pythonfor-loopmatplotlibcolorbar

How to evenly spread ticks on a colorbar?


I have been trying to make a plot with some evenly spaced tick in my colorbar, but so far my results always give me a colorbar with the distance between the ticks proportional to their values as shown in the image below:

import numpy as np
import matplotlib
import matplotlib as plt


T= [0.01, 0.02, 0.03, 0.04] #values for the colourbar to use in equation in for loop
x=np.linspace[0, 8, 100]
e=1/(np.exp(x)+1)       #factor used in equation dependent on the x-axis values
a=6.4*10**(-9)         
b= 1.51                  # constants for the equation

pof6= [number **6 for number in T]

norm = matplotlib.colors.Normalize(vmin=np.min(pof6), vmax=np.max(pof6))  #colourbar max and min values
c_m = matplotlib.cm.cool
s_m = matplotlib.cm.ScalarMappable(cmap='jet', norm=norm)

s_m.set_array([])

#below is the for loop that uses one value of T at a time, represented as t in the equation


for t in pof6:            
    plt.plot(x, b*x/(((a*t*x**2/(m**2))+1)**2)*e, color=s_m.to_rgba(t)) 

func = lambda x,pos: "{:g}".format(x)
fmt = matplotlib.ticker.FuncFormatter(func)

c_bar=plt.colorbar(s_m, format=fmt, ticks=[0.01**6,0.02* 0.03**6, 0.04**6])

plt.legend()
plt.xlabel('y=E/T')
plt.ylabel('$f_{ν_s}$')
c_bar.set_label(r'T(K)')
plt.show()

I have attempted applying some of the solutions suggested here n=on the website, like Spread custom tick labels evenly over colorbar but haven't been successful at that.

enter image description here


Solution

  • You're using a linear norm, where the pof values are very close to each other. It helps to use a LogNorm. The tick formatter can be adapted to show the values in their **6 format.

    The code below shifts the four functions a bit, because with the code from the example all plots seem to coincide. At least when I use something like m=2 (m is not defined in the code).

    import numpy as np
    import matplotlib.pyplot as plt
    from matplotlib import colors as mcolors
    from matplotlib import ticker as mticker
    
    T = [0.01, 0.02, 0.03, 0.04]  # values for the colourbar to use in equation in for loop
    x = np.linspace(0, 8, 100)
    e = 1 / (np.exp(x) + 1)  # factor used in equation dependent on the x-axis values
    a = 6.4 * 10 ** (-9)
    b = 1.51  # constants for the equation
    
    pof6 = [number ** 6 for number in T]
    
    norm = mcolors.LogNorm(vmin=np.min(pof6), vmax=np.max(pof6))  # colourbar max and min values
    s_m = plt.cm.ScalarMappable(cmap='jet', norm=norm)
    s_m.set_array([])
    
    m = 2
    for t in pof6:
        plt.plot(x, b * x / (((a * t * x ** 2 / (m ** 2)) + 1) ** 2) * e + 10*t**(1/6), color=s_m.to_rgba(t))
    
    func = lambda x, pos: "{:g}**6".format(x**(1/6) )
    fmt = mticker.FuncFormatter(func)
    
    c_bar = plt.colorbar(s_m, format=fmt, ticks=pof6)
    c_bar.set_label(r'T(K)')
    
    # plt.legend() # there are no labels set, so a default legend can't be created
    plt.xlabel('y=E/T')
    plt.ylabel('$f_{ν_s}$')
    plt.show()
    

    example plot

    If you want a legend, you need to put a label to each curve, for example:

    for t in pof6:
        plt.plot(x, b * x / (((a * t * x ** 2 / (m ** 2)) + 1) ** 2) * e, color=s_m.to_rgba(t),
                 label=f'$t = {t**(1/6):g}^6$')
    plt.legend()