Как спроецировать контурную карту на плоскость xy с помощью matplotlib? ⇐ Python

[Anonymous]

Как построить контурную карту на плоскости xy, а не на трехмерной поверхности функции?

Я хотел построить контур там, где z = 0.

Однако мои текущие результаты дают следующее:
[img]https://i.sstatic.net /gXRFoDIz.png[/img]


Следующий код является моей реализацией:
import numpy as np
import matplotlib.pyplot as plt

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

x = np.linspace(-10,10,100)
y = np.linspace(-10,10,100)
X,Y = np.meshgrid(x,y) # array of values containing all value pairs (xi, yi)
Z = f(X,Y)

fig = plt.figure()
ax = plt.axes(projection='3d')
ax.plot_surface(X, Y, Z, cmap="autumn_r", rstride=1, cstride=1)
ax.contour(X, Y, Z, 10, cmap="autumn_r", linestyles="solid")
plt.show()


Подробнее здесь: https://stackoverflow.com/questions/787 ... matplotlib