我想使用numpy的布尔索引,在绘图时仅绘制函数满足特定条件-sigma < 0的区域,如下所示的代码。当我尝试像这个例子中change_line_style那样使用numpy的布尔索引时,我会遇到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()