Python matplotlib контурная диаграмма логарифмическая цветовая гамма - PullRequest
Новая
       9

Python matplotlib контурная диаграмма логарифмическая цветовая гамма

10 голосов
/ 21 апреля 2011

У меня проблемы с контурным графиком, использующим логарифмическое масштабирование цвета.Я хочу указать уровни вручную.Matplotlib, однако, рисует цветовую полосу странным образом - метки размещены неправильно и появляется только один цвет.Идея основана на http://adversus.110mb.com/?cat=8

Есть ли кто-нибудь, кто может мне помочь?Я использую последнюю версию git-repository matplotlib, v1.1.0 (2011-04-21) 

import matplotlib.pyplot as plt
import numpy as np
from matplotlib.mlab import bivariate_normal
from matplotlib.colors import LogNorm
from matplotlib.backends.backend_pdf import PdfPages


delta = 0.5

x = np.arange(-3.0, 4.001, delta)
y = np.arange(-4.0, 3.001, delta)
X, Y = np.meshgrid(x, y)
Z = bivariate_normal(X, Y, 1.0, 1.0, 0.0, 0.0)

fig  = plt.figure()
ax   = fig.add_subplot(1,1,1)
#axim = ax.imshow(Z, norm = LogNorm())
axim    = ax.contourf(X,Y,Z,levels=[1e0,1e-1,1e-2,1e-3],cmap=plt.cm.jet,norm = LogNorm())
cb   = fig.colorbar(axim)

pp = PdfPages('fig.pdf')
pp.savefig()
pp.close()


plt.show()

Большое спасибо за вашу помощь!Он отлично работает, как вы предложили ... Однако у меня есть еще один вопрос: почему matplotlib не позволяет мне выбирать количество линий уровня в логарифмическом режиме: 

import matplotlib.pyplot as plt
import numpy as np
from matplotlib.mlab import bivariate_normal
from matplotlib.colors import LogNorm
from matplotlib.backends.backend_pdf import PdfPages


delta = 0.5

x = np.arange(-3.0, 4.001, delta)
y = np.arange(-4.0, 3.001, delta)
X, Y = np.meshgrid(x, y)
Z = bivariate_normal(X, Y, 1.0, 1.0, 0.0, 0.0)

fig  = plt.figure()
ax   = fig.add_subplot(1,1,1)
#axim = ax.imshow(Z, norm = LogNorm())
#axim   = ax.contourf(X,Y,Z,levels=[1e-3,1e-2,1e-1,1e0],cmap=plt.cm.jet,norm = LogNorm())
axim    = ax.contourf(X,Y,Z,20,cmap=plt.cm.jet,norm = LogNorm())
cb   = fig.colorbar(axim)

pp = PdfPages('fig.pdf')
pp.savefig()
pp.close()


plt.show()

http://i.stack.imgur.com/VeVFQ.png

Это была моя первоначальная проблема ...

Ответы [ 2 ]

11 голосов
/ 21 апреля 2011

Так что это легко исправить;Ваш порядок уровней означает, что самый низкий уровень рисуется последним и поэтому охватывает все!Попробуйте: 

axim    = ax.contourf(X,Y,Z,levels=[1e-3, 1e-2, 1e-1, 1e0],cmap=plt.cm.jet,norm = LogNorm())

, и вы получите желаемый результат.

2 голосов
/ 21 апреля 2011

Похоже, levels ожидает увеличения значений. Попробуйте изменить их на: levels=[1e-3, 1e-2, 1e-1, 1e0] и посмотрите, решит ли это вашу проблему.

...