Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Cách vẽ bản đồ mật độ 3D trong Python với Matplotlib

Để vẽ bản đồ mật độ 3D:

  • Tạo side, x, y, z bằng numpy. linspace tạo dữ liệu giữa hai điểm.
  • Trả về ma trận tọa độ từ vector tọa độ dùng side.
  • Tạo dữ liệu mũ dùng x, y (bước 2).
  • Vẽ pseudocolor với lưới chữ nhật không đều bằng pcolormesh().
  • Hiển thị bằng show().

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()

Kết quả

Cách vẽ bản đồ mật độ 3D trong Python với Matplotlib