Comment modifier les labels d'une colorbar avec imshow sous matplotlib ?

Exemples de comment personnaliser les labels de la colorbar avec imshow de matplotib:

Colorbar de départ

Tracer une simple colorbar avec matplotlib:

import numpy as np
import matplotlib.pyplot as plt

def f(x,y):
    return (x+y)*np.exp(-5.0*(x**2+y**2))

x,y = np.mgrid[-1:1:100j, -1:1:100j]

z = f(x,y)

plt.imshow(z,extent=[-1,1,-1,1])

plt.colorbar()

plt.savefig("ImshowColorBar01.png")

plt.show()

Modifier la taille des labels

Pour modifier la taille des labels on peut utiliser l'option labelsize, exemple diminuer la taille des labels labelsize=7:

import numpy as np
import matplotlib.pyplot as plt

def f(x,y):
    return (x+y)*np.exp(-5.0*(x**2+y**2))

x,y = np.mgrid[-1:1:100j, -1:1:100j]

z = f(x,y)

plt.imshow(z,extent=[-1,1,-1,1])

cb = plt.colorbar()

cb.ax.tick_params(labelsize=7)

plt.savefig("ImshowColorBar02.png")

plt.show()

Modifier la position des labels

Pour modifier la position des labels on peut utiliser l'argument ticks, exemple

import numpy as np
import matplotlib.pyplot as plt

def f(x,y):
    return (x+y)*np.exp(-5.0*(x**2+y**2))

x,y = np.mgrid[-1:1:100j, -1:1:100j]

z = f(x,y)

plt.imshow(z,extent=[-1,1,-1,1],vmin=z.min(),vmax=z.max())

v1 = np.linspace(z.min(), z.max(), 8, endpoint=True)
cb = plt.colorbar(ticks=v1)

plt.savefig("ImshowColorBar03.png")

plt.show()

Modifier le format des labels

On peut aussi changer le format des labels:

import numpy as np
import matplotlib.pyplot as plt

def f(x,y):
    return (x+y)*np.exp(-5.0*(x**2+y**2))

x,y = np.mgrid[-1:1:100j, -1:1:100j]

z = f(x,y)

f = plt.figure()
ax = f.add_subplot(111)

plt.imshow(z,extent=[-1,1,-1,1],vmin=z.min(),vmax=z.max())

v1 = np.linspace(z.min(), z.max(), 8, endpoint=True)
cb = plt.colorbar(ticks=v1)
cb.ax.set_yticklabels(["{:4.2f}".format(i) for i in v1])

plt.savefig("ImshowColorBar04.png")

Références