Матплотлиб - Контурный график

Контурные графики (иногда называемые графиками уровней) - это способ показать трехмерную поверхность на двухмерной плоскости. Он отображает две переменные-предикторы XY по оси Y и переменную отклика Z в виде контуров. Эти контуры иногда называют z-срезами или значениями изоотклика.

Контурный график подходит, если вы хотите увидеть, как значение Z изменяется в зависимости от двух входов X и Y, так что Z = f (X, Y). Линия изолиний или изолинии функции двух переменных - это кривая, вдоль которой функция имеет постоянное значение.

Независимые переменные x и y обычно ограничиваются регулярной сеткой, называемой сеткой. Numpy.meshgrid создает прямоугольную сетку из массива значений x и массива значений y.

Matplotlib API содержит функции contour () и contourf (), которые рисуют контурные линии и контуры с заливкой соответственно. Обе функции требуют трех параметров x, y и z.

import numpy as np
import matplotlib.pyplot as plt
xlist = np.linspace(-3.0, 3.0, 100)
ylist = np.linspace(-3.0, 3.0, 100)
X, Y = np.meshgrid(xlist, ylist)
Z = np.sqrt(X**2 + Y**2)
fig,ax=plt.subplots(1,1)
cp = ax.contourf(X, Y, Z)
fig.colorbar(cp) # Add a colorbar to a plot
ax.set_title('Filled Contours Plot')
#ax.set_xlabel('x (cm)')
ax.set_ylabel('y (cm)')
plt.show()