Làm thế nào để vẽ một bản đồ mật độ trong Python Matplotlib?

Để vẽ một bản đồ mật độ trong Python, chúng ta có thể thực hiện các bước sau -

• Tạo bên, x, y và z bằng cách sử dụng numpy. Numpy linspace giúp tạo dữ liệu giữa hai điểm dựa trên số thứ ba.

• Trả về ma trận tọa độ từ vectơ tọa độ bằng cách sử dụng dữ liệu bên.

• Tạo dữ liệu theo cấp số nhân bằng cách sử dụng x và y (Bước 2).

• Tạo biểu đồ màu giả với lưới hình chữ nhật không thông thường bằng cách sử dụng pcolormesh () phương pháp.

• Để hiển thị hình này, hãy sử dụng show () phương pháp.

Ví dụ

from matplotlib import pyplot as plt, cm, colors
import numpy as np

plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True

side = np.linspace(-2, 2, 15)
X, Y = np.meshgrid(side, side)
Z = np.exp(-((X - 1) ** 2 + Y ** 2))

plt.pcolormesh(X, Y, Z, shading='auto')

plt.show()

Đầu ra