Jak utworzyć wykres konturowy w matplotlib

Wykres konturowy to rodzaj wykresu, który umożliwia wizualizację danych trójwymiarowych w dwóch wymiarach za pomocą konturów.

Możesz utworzyć wykres konturowy w Matplotlib za pomocą następujących dwóch funkcji:

• matplotlib.pyplot.contour() – Tworzy wykresy konturowe.
• matplotlib.pyplot.contourf() – Tworzy wypełnione wykresy konturowe.

Poniższe przykłady pokazują, jak w praktyce wykorzystać te dwie funkcje.

Przykład 1: Wykres konturowy w Matplotlib

Załóżmy, że mamy w Pythonie następujące dane:

 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)

Możemy użyć następującego kodu, aby utworzyć wykres konturowy dla danych:

 import matplotlib.pyplot as plt

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

Gdy na wykresie używany jest tylko jeden kolor, linie przerywane przedstawiają wartości ujemne, a linie ciągłe przedstawiają wartości dodatnie.

Alternatywą jest określenie mapy kolorów za pomocą argumentu cmap . Za pomocą argumentu poziomów możemy również określić więcej linii do wykorzystania na wykresie:

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

Zdecydowaliśmy się użyć cmap „Reds”, ale pełną listę opcji palety kolorów można znaleźć na stronie dokumentacji Matplotlib .

Przykład 2: Wypełniony wykres konturowy w Matplotlib

Wypełniona ścieżka konturowa jest podobna do ścieżki konturowej, z tą różnicą, że przestrzenie pomiędzy liniami są wypełnione.

Poniższy kod pokazuje, jak używać funkcji konturf() do tworzenia wypełnionego wykresu konturowego dla tych samych danych, co w poprzednim przykładzie:

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

Możemy również użyć funkcji colorbar() , aby dodać oznaczony pasek koloru obok wykresu:

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

Więcej samouczków dotyczących Matplotlib znajdziesz tutaj .