matplotlib tracé de contour: proportionnelle barre de couleur niveaux dans l'échelle logarithmique

Serait-il possible d'avoir des niveaux d'la barre de couleur en échelle logarithmique, comme dans l'image ci-dessous?

matplotlib tracé de contour: proportionnelle barre de couleur niveaux dans l'échelle logarithmique

Voici un exemple de code où il pourrait être mis en œuvre:

import matplotlib.pyplot as plt
import numpy as np
from matplotlib.colors import LogNorm
delta = 0.025

x = y = np.arange(0, 3.01, delta)
X, Y = np.meshgrid(x, y)
Z1 = plt.mlab.bivariate_normal(X, Y, 1.0, 1.0, 0.0, 0.0)
Z2 = plt.mlab.bivariate_normal(X, Y, 1.5, 0.5, 1, 1)
Z = 1e6 * (Z1* Z2)

fig=plt.figure()
ax1 = fig.add_subplot(111)
lvls = np.logspace(0,4,20)
CF = ax1.contourf(X,Y,Z,
         norm = LogNorm(),
         levels = lvls
        )
CS = ax1.contour(X,Y,Z,
         norm = LogNorm(),
         colors = 'k',
         levels = lvls
        )
cbar = plt.colorbar(CF, ticks=lvls, format='%.4f')
plt.show()

matplotlib tracé de contour: proportionnelle barre de couleur niveaux dans l'échelle logarithmique

Je suis à l'aide de python 2.7.3 avec matplotlib 1.1.1 sur Windows 7.

  • Votre barre de couleur a déjà une échelle logarithmique.
  • Je crois que les OP se demande comment définir la tique locator et formateur sur la barre de couleur pour afficher les étiquettes régulièrement des intervalles d'enregistrement.
  • Ah, il me semblait rappeler que les OP dans son question voulait le les graduations pour être placé à l'valeurs données dans le tableau lvls avec logarithmique espacement, mais j'ai peut-être misundertood ce qu'il voulait dire. Merci pour ça.
  • Vous pouvez probablement faire quelque chose avec matplotlib.colors.LogNorm() similaire à cette réponse
InformationsquelleAutor user2145054 | 2013-08-12
  1. 13

    Je propose de générer un pseudo barre de couleur comme suit (voir les commentaires pour plus d'explications): 

    import matplotlib.pyplot as plt
import numpy as np
from matplotlib.colors import LogNorm
import matplotlib.gridspec as gridspec
delta = 0.025
x = y = np.arange(0, 3.01, delta)
X, Y = np.meshgrid(x, y)
Z1 = plt.mlab.bivariate_normal(X, Y, 1.0, 1.0, 0.0, 0.0)
Z2 = plt.mlab.bivariate_normal(X, Y, 1.5, 0.5, 1, 1)
Z = 1e6 * (Z1 * Z2)
fig=plt.figure()
#
# define 2 subplots, using gridspec to control the 
# width ratios:
#
# note: you have to import matplotlib.gridspec for this
#
gs = gridspec.GridSpec(1, 2,width_ratios=[15,1])
# the 1st subplot
ax1 = plt.subplot(gs[0])
lvls = np.logspace(0,4,20)
CF = ax1.contourf(X,Y,Z,
norm = LogNorm(),
levels = lvls
)
CS = ax1.contour(X,Y,Z,
norm = LogNorm(),
colors = 'k',
levels = lvls
)
#
# the pseudo-colorbar
#
# the 2nd subplot
ax2 = plt.subplot(gs[1])        
#
# new levels!
#
# np.logspace gives you logarithmically spaced levels - 
# this, however, is not what you want in your colorbar
#
# you want equally spaced labels for each exponential group:
#
levls = np.linspace(1,10,10)
levls = np.concatenate((levls[:-1],np.linspace(10,100,10)))
levls = np.concatenate((levls[:-1],np.linspace(100,1000,10)))
levls = np.concatenate((levls[:-1],np.linspace(1000,10000,10)))
#
# simple x,y setup for a contourf plot to serve as colorbar
#
XC = [np.zeros(len(levls)), np.ones(len(levls))]
YC = [levls, levls]
CM = ax2.contourf(XC,YC,YC, levels=levls, norm = LogNorm())
# log y-scale
ax2.set_yscale('log')  
# y-labels on the right
ax2.yaxis.tick_right()
# no x-ticks
ax2.set_xticks([])
plt.show()

    Cela vous donnera une intrigue comme ceci:

    matplotlib tracé de contour: proportionnelle barre de couleur niveaux dans l'échelle logarithmique

    MODIFIER

    Ou, utiliser quelque chose qui ressemble à de nouveaux niveaux et de la spacing='proportional' option lors de l'appel de la colorbar:

    1. remplacer cette ligne: 

      lvls = np.logspace(0,4,20)

      avec ces: 

      lvls = np.linspace(1,10,5)
lvls = np.concatenate((lvls[:-1],np.linspace(10,100,5)))
lvls = np.concatenate((lvls[:-1],np.linspace(100,1000,5)))
lvls = np.concatenate((lvls[:-1],np.linspace(1000,10000,5)))

    2. remplacer cette ligne: 

      cbar = plt.colorbar(CF, ticks=lvls, format='%.4f')

      avec ceci: 

      cbar = plt.colorbar(CF, ticks=lvls, format='%.2f', spacing='proportional')

    Et vous allez vous retrouver avec cette intrigue:

    matplotlib tracé de contour: proportionnelle barre de couleur niveaux dans l'échelle logarithmique

    (le format a été changé, parce que la nouvelle tiques ne nécessitent pas de 4 décimales)

    EDIT 2

    Si vous voulais pour générer automatiquement les niveaux, comme ceux que j'ai utilisé, vous pouvez considérer ce morceau de code: 

    levels = []
LAST_EXP = 4
N_LEVELS = 5
for E in range(0,LAST_EXP):
levels = np.concatenate((levels[:-1],np.linspace(10**E,10**(E+1),N_LEVELS)))
    InformationsquelleAutor Schorsch