如何使用不同颜色的等高线图覆盖controuf图?

时间:2014-12-11 08:50:23

标签: python matplotlib contour colormap contourf

(我问过the same in MATLAB before

我想覆盖例如seismic - cmapped contourf - 情节(或pcolor),gray比例contour - 情节,但是当我添加后者时,它也会改变以前的色彩映射。如何解决这个问题?

1 个答案:

答案 0 :(得分:2)

这个答案几乎全部来自contour demo example

import numpy as np
import matplotlib.cm as cm
import matplotlib.mlab as mlab # for setting up the data
import matplotlib.pyplot as plt

# set up example data:
delta = 0.025
x = np.arange(-3.0, 3.0, delta)
y = np.arange(-2.0, 2.0, delta)
X, Y = np.meshgrid(x, y)
Z1 = mlab.bivariate_normal(X, Y, 1.0, 1.0, 0.0, 0.0)
Z2 = mlab.bivariate_normal(X, Y, 1.5, 0.5, 1, 1)
# difference of Gaussians
Z = 10.0 * (Z2 - Z1)

levels = 10

# plot the filled contour
# using a colormap (jet)
CF = plt.contourf(Z, levels,
                  extent=(-3,3,-2,2),cmap=cm.jet)

# plot the contour lines
# using gray scale
CL = plt.contour(Z, levels,
                 linewidths=2,
                 extent=(-3,3,-2,2),cmap=cm.gray)

# plot color bars for both contours (filled and lines)
CB = plt.colorbar(CL, extend='both')
CBI = plt.colorbar(CF, orientation='horizontal')

# Plotting the second colorbar makes 
# the original colorbar look a bit out of place,
# so let's improve its position.

l,b,w,h = plt.gca().get_position().bounds
ll,bb,ww,hh = CB.ax.get_position().bounds
CB.ax.set_position([ll, b, ww, h])

plt.show()

你最终会得到这个情节:

two color scales