Условный контур сюжета

Question

Код ниже - это то, с чем я играю в минуту:

x = np.linspace(0,30,1000)
y = np.linspace(0,30,1000)
X,Y = np.meshgrid(x,y)

def f(x,y):
    return x**2 + y**2

Z = f(X,Y)


plt.contour(X, Y, Z, colors='black');

Я хочу, чтобы на этом графике отображалась какая-то запрещенная область, например, когда f (x, y) <9;Я хочу, чтобы это было затенено и добавлено в сюжет.Как именно я это сделаю? </p>

Я пытался использовать plt.contourf, но не могу заставить его работать.

Answer 1

Я думаю, вы можете сделать это таким образом, используя contourf, используйте контур для заливки сплошным красным цветом, а затем замаскируйте область, которую вы хотите отобразить, с помощью вашей контурной диаграммы:

x = np.linspace(0,30,1000)
y = np.linspace(0,30,1000)
X,Y = np.meshgrid(x,y)

def f(x,y):
    return x**2 + y**2

Z = f(X,Y)

d = np.ma.array(Z, mask=Z>9)

plt.contour(X, Y, Z, colors='black')
plt.contourf(X, Y, d, colors='red');

Выход: