Cara membuat plot kontur di matplotlib

Plot kontur adalah jenis plot yang memungkinkan kita memvisualisasikan data tiga dimensi dalam dua dimensi menggunakan kontur.

Anda dapat membuat plot kontur di Matplotlib menggunakan dua fungsi berikut:

• matplotlib.pyplot.contour() – Membuat plot kontur.
• matplotlib.pyplot.contourf() – Membuat plot kontur terisi.

Contoh berikut menunjukkan cara menggunakan kedua fungsi ini dalam praktik.

Contoh 1: Plot kontur di Matplotlib

Misalkan kita memiliki data berikut dengan 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)

Kita dapat menggunakan kode berikut untuk membuat plot kontur untuk data:

 import matplotlib.pyplot as plt

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

Jika hanya satu warna yang digunakan untuk plot, garis putus-putus mewakili nilai negatif dan garis padat mewakili nilai positif.

Alternatifnya adalah menentukan peta warna menggunakan argumen cmap . Kita juga dapat menentukan lebih banyak baris untuk digunakan dalam plot dengan argumen level :

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

Kami memilih untuk menggunakan cmap ‘Merah’ tetapi Anda dapat menemukan daftar lengkap opsi palet warna di halaman dokumentasi Matplotlib .

Contoh 2: Plot Kontur Terisi di Matplotlib

Jalur kontur terisi mirip dengan jalur kontur, hanya saja ruang antar garisnya terisi.

Kode berikut menunjukkan cara menggunakan fungsi contourf() untuk membuat plot kontur terisi untuk data yang sama dengan yang digunakan pada contoh sebelumnya:

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

Kita juga bisa menggunakan fungsi colorbar() untuk menambahkan bilah warna berlabel di sebelah plot:

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

Anda dapat menemukan lebih banyak tutorial Matplotlib di sini .