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

Để vẽ bản đồ mật độ 3D bằng Python với matplotlib, 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