python matplotlib - 等高线图 - 置信区间

Question

我试图用matplotlib.pyplot.contour绘制数据网格上的轮廓（可行），但将等高线放置在距离峰值1,2和3 sigma处。除了蛮力之外，有没有一种简洁的方式来做到这一点？谢谢！

Python版本是

Python 2.7.2 | EPD 7.2-2（64位）| （默认，2011年9月7日，16:31:15） [GCC 4.0.1（Apple Inc. build 5493）]在达尔文

Answer 1

您可以指定绘制轮廓的列表z-values。所以你所要做的就是为你的发行收集正确的z-values。下面是 '1，2，和3西格玛从峰值远离' 的示例：

代码：

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()