web-dev-qa-db-fra.com

Python / Matplotlib - Plage de barres de couleurs et valeurs d'affichage

Lorsque j'utilise matplotlib avec un tracé de contour, j'ai du mal à afficher la barre de couleurs comme je le souhaite. J'ai lu de nombreux exemples similaires, mais je n'ai toujours pas pu obtenir ce que je veux.

Dans l'image ci-dessous, je veux que deux choses changent. Je veux que la valeur minimale et les valeurs maximales soient affichées sur la barre de couleur (le maximum doit être 2,0 et le min -0,1). Ces deux valeurs doivent être au bord même de la barre de couleur. De plus, je veux que la barre de couleurs affiche la valeur à chaque transition de couleur. Par exemple. dans le graphique ci-dessous, entre 2.1 et 1.8, il y a une autre transition de couleur où la valeur n'est pas affichée.

Est-ce que quelqu'un peut m'aider s'il vous plait? Je pense que je devrai peut-être utiliser la norme, mais cela n'a pas fonctionné pour moi jusqu'à présent.

Merci,

enter image description here

Code:

import numpy as np
import matplotlib.pyplot as plt

xi = np.array([0., 0.5, 1.0])
yi = np.array([0., 0.5, 1.0])
zi = np.array([[0., 1.0, 2.0],
               [0., 1.0, 2.0],
               [-0.1, 1.0, 2.0]])

plt.contour(xi, yi, zi, 15, linewidths=0.5, colors='k')
plt.contourf(xi, yi, zi, 15, cmap=plt.cm.jet)
plt.colorbar()
plt.show()
pythonmatplotlibcontourcolorbar
28
Scott B

Si je comprends bien ce que vous voulez, je pense que cela devrait le faire:

import numpy as np
import matplotlib.pyplot as plt

xi = np.array([0., 0.5, 1.0])
yi = np.array([0., 0.5, 1.0])
zi = np.array([[0., 1.0, 2.0],
               [0., 1.0, 2.0],
               [-0.1, 1.0, 2.0]])

v = np.linspace(-.1, 2.0, 15, endpoint=True)
plt.contour(xi, yi, zi, v, linewidths=0.5, colors='k')
plt.contourf(xi, yi, zi, v, cmap=plt.cm.jet)
x = plt.colorbar(ticks=v)
print x
plt.show()

enter image description here

37
tom10