Як створити контурну діаграму в matplotlib

Контурна діаграма — це тип діаграми, який дозволяє візуалізувати тривимірні дані у двох вимірах за допомогою контурів.

Ви можете створити контурний графік у Matplotlib за допомогою наступних двох функцій:

• matplotlib.pyplot.contour() – Створює контурні графіки.
• matplotlib.pyplot.contourf() – Створює контурні графіки із заливкою.

Наступні приклади показують, як використовувати ці дві функції на практиці.

Приклад 1: контурний графік у Matplotlib

Припустімо, що ми маємо такі дані в Python:

 import numpy as np

x = np.linspace(0, 5, 50)
y = np.linspace(0, 5, 40)

X, Y = np.meshgrid(x, y)
Z = np.sin(X*2+Y)*3 + np.cos(Y+5)

Ми можемо використати такий код, щоб створити контурний графік для даних:

 import matplotlib.pyplot as plt

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

Якщо для графіка використовується лише один колір, пунктирні лінії позначають від’ємні значення, а суцільні — додатні.

Альтернативою є визначення карти кольорів за допомогою аргументу cmap . Ми також можемо вказати більше рядків для використання на графіку за допомогою аргументу рівнів :

 plt. contour (X, Y, Z, levels= 30 , cmap=' Reds ') 

Ми вирішили використовувати cmap «Червоні», але ви можете знайти повний список параметрів палітри кольорів на сторінці документації Matplotlib .

Приклад 2: контур із заливкою в Matplotlib

Контур із заливкою подібний до контуру, за винятком того, що проміжки між лініями заповнені.

У наведеному нижче коді показано, як використовувати функцію contourf() для створення контурної діаграми із заливкою для тих самих даних, що використовуються в попередньому прикладі:

 plt. contourf (X, Y, Z, cmap=' Reds ') 

Ми також можемо використовувати функцію colorbar() , щоб додати кольорову смугу з мітками поруч із графіком:

 plt. contourf (X, Y, Z, cmap=' Reds ')
plt. colorbar () 

Ви можете знайти більше навчальних посібників Matplotlib тут .