Définition explicite de bacs de barres de couleurs dans plt.scatter

Question

J'ai créé un diagramme de dispersion sur une carte de base des États-Unis en utilisant pyplot.scatter dans lequel les segments de la barre de couleur sont régulièrement espacés sur la plage des données d'entrée. Cependant, je voudrais contrôler explicitement les cases de la barre de couleur afin de créer des cases inégales, par ex. [-20, -10, -5,-4,-3,-2,-1,0,1,2,3,4,5,50] au lieu de bacs espacés uniformément entre -20 et 50. Comment cela peut-il être fait?

Answer 1

Pour définir vos propres tiques dans un colorbar en plt.scatter vous pouvez modifier le paramètre ticks dans fig.colorbar à ce que la liste des valeurs que vous désirez:

import matplotlib.pyplot as plt 
fig, ax = plt.subplots() 
cm = plt.cm.get_cmap('RdYlBu') 
xy = range(20) 
z = xy 
sc = plt.scatter(xy, xy, c=z, vmin=0, vmax=20, s=35, cmap=cm) 

cbar = fig.colorbar(sc, ticks=[0, 1, 15]) 
plt.show()