Question

我想使用numpy的布尔索引来仅绘制我满足某个条件的函数的区域 - sigma＆lt; 0，如下面的代码所示。当我尝试使用numpy的布尔索引时，如本例所示 - 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()

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

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

1 个答案: