Cách trực quan hóa dữ liệu scalar 2D với Matplotlib?

Để trực quan hóa dữ liệu scalar 2D với matplotlib, chúng ta có thể thực hiện các bước sau:

• Thiết lập kích thước figure và điều chỉnh padding giữa và xung quanh subplots.
• Khởi tạo biến N cho mẫu dữ liệu.
• Tạo điểm dữ liệu x và y bằng numpy.
• Lấy ma trận tọa độ từ vector tọa độ.
• Lấy điểm dữ liệu z bằng numpy.
• Tạo biểu đồ pseudocolor với lưới chữ nhật không đều.
• Để hiển thị figure, dùng phương thức show().

Ví dụ

import numpy as np
from matplotlib import pyplot as plt

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

n = 256
x = np.linspace(-3., 3., n)
y = np.linspace(-3., 3., n)
X, Y = np.meshgrid(x, y)
Z = X * np.sinc(X ** 2 + Y ** 2)

plt.pcolormesh(X, Y, Z, cmap='copper', shading='flat')
plt.show()

Kết quả