Matplotlib 挑战 - 改变等高线图中的线条样式答案

【问题标题】：Matplotlib challenge - changing line style in contour mapMatplotlib 挑战 - 改变等高线图中的线条样式
【发布时间】：2014-11-30 13:32:40
【问题描述】：

我想使用 numpy 的布尔索引仅绘制满足某个条件的函数区域 - sigma change_line_style，我得到 TypeError。有没有办法使用 numpy 的布尔索引来使用轮廓函数仅绘制某些区域？

import numpy as np
import matplotlib.pyplot as plt
from matplotlib import rc
rc('font',**{'family':'sans-serif','sans-serif':['Helvetica']})
rc('text', usetex=True)

plt.figure()
plt.ylim([-2,2])
plt.xlim([-0.5,0.5])

plt.xlabel(r'\lambda')
plt.ylabel(r'u')

lambda_x = np.linspace(-0.5,0.5,1000)
u = np.linspace(-2,2,1000)

X,Y = np.meshgrid(lambda_x,u)
# Drawing f=0 lines for implicit function f(u;lambda) = lambda*u  + u**3 - u**5
f1 = X  + Y**2 - Y**4
f2 = Y

sigma = X + 3*Y**2 - 5*Y**4

stable = sigma < 0
#plt.setp(zc, linewidth=4)
print stable.shape
print f1.shape

plt.contour(X[stable],Y[stable], f1[stable], levels = [0],colors = ('r'),linewidths = 4,extend='both')
plt.contour(X[stable],Y[stable], f2[stable], levels = [0],colors = ('b'),linewidths = 4,extend='both')

plt.show()

【问题讨论】：

不能提前准备数据吗？
我在这里找不到最小的例子。

标签： python matplotlib contour

【解决方案1】：

知道了！

我使用了 numpy.ma.masked_where() 函数，如这段代码：

import numpy as np
import matplotlib.pyplot as plt
from matplotlib import rc
rc('font',**{'family':'sans-serif','sans-serif':['Helvetica']})
rc('text', usetex=True)

plt.figure()
plt.ylim([-2,2])
plt.xlim([-0.5,0.5])

lambda_x = np.linspace(-0.5,0.5,1000)
u = np.linspace(-2,2,1000)

X,Y = np.meshgrid(lambda_x,u)
# Drawing f=0 lines for implicit function f(u;lambda) = lambda*u  + u**3 - u**5
f1 = X  + Y**2 - Y**4
f2 = Y

sigma = X + 3*Y**2 - 5*Y**4

unstable = sigma > 0
stable = sigma < 0

#for i in range(0,10):
    #f1[i] = f1[i][stable[i]]

plt.contour(np.ma.masked_where(unstable, X),np.ma.masked_where(unstable, Y), np.ma.masked_where(unstable, f1), levels = [0],colors = ('r'),linewidths = 4,extend='both')
plt.contour(np.ma.masked_where(stable, X),np.ma.masked_where(stable, Y), np.ma.masked_where(stable, f1), levels = [0],colors = ('b'),linewidths = 4,linestyles = 'dashed',extend='both')
plt.contour(np.ma.masked_where(unstable, X),np.ma.masked_where(unstable, Y), np.ma.masked_where(unstable, f2), levels = [0],colors = ('r'),linewidths = 4,extend='both')
plt.contour(np.ma.masked_where(stable, X),np.ma.masked_where(stable, Y), np.ma.masked_where(stable, f2), levels = [0],colors = ('b'),linewidths = 4,linestyles = 'dashed',extend='both')

plt.show()

【讨论】：