matplotlib python - diagrama de contorno - intervalos de confianza

Question

Estoy tratando de trazar contornos (factibles) en una cuadrícula de datos usando matplotlib.pyplot.contour, pero con los contornos colocados a 1, 2 y 3 sigma lejos del valor máximo . ¿Hay una forma ordenada de hacer esto aparte de la fuerza bruta? ¡Gracias!

versión

Python es

Python 2.7.2 | EPD 7.2-2 (64 bits) | (predeterminado, 7 de septiembre de 2011, 16:31:15) [GCC 4.0.1 (Apple Inc. compilación 5493)] en darwin

Answer 1

Puede especificar una lista de z-values donde se dibujan los contornos. Entonces, todo lo que tiene que hacer es recopilar el z-values correcto para su distribución. Este es un ejemplo de '1, 2, y 3 sigma lejos del valor de pico':

Código:

import numpy as np 
import matplotlib.cm as cm 
import matplotlib.mlab as mlab 
import matplotlib.pyplot as plt 

#Set up the 2D Gaussian: 
delta = 0.025 
x = np.arange(-3.0, 3.0, delta) 
y = np.arange(-3.0, 3.0, delta) 
X, Y = np.meshgrid(x, y) 
sigma = 1.0 
Z = mlab.bivariate_normal(X, Y, sigma, sigma, 0.0, 0.0) 
#Get Z values for contours 1, 2, and 3 sigma away from peak: 
z1 = mlab.bivariate_normal(0, 1 * sigma, sigma, sigma, 0.0, 0.0) 
z2 = mlab.bivariate_normal(0, 2 * sigma, sigma, sigma, 0.0, 0.0) 
z3 = mlab.bivariate_normal(0, 3 * sigma, sigma, sigma, 0.0, 0.0) 

plt.figure() 
#plot Gaussian: 
im = plt.imshow(Z, interpolation='bilinear', origin='lower', 
       extent=(-50,50,-50,50),cmap=cm.gray) 
#Plot contours at whatever z values we want: 
CS = plt.contour(Z, [z1, z2, z3], origin='lower', extent=(-50,50,-50,50),colors='red') 
plt.savefig('fig.png') 
plt.show()