Làm thế nào để vẽ một ma trận 2D trong Python với Matplotlib trên thanh màu?

Để vẽ ma trận 2D bằng Python với thanh màu, chúng ta có thể sử dụng numpy để tạo ma trận mảng 2D và sử dụng ma trận đó trong imshow () phương pháp.

Các bước

• Tạo data2D sử dụng numpy.

• Sử dụng imshow () phương pháp hiển thị dữ liệu dưới dạng hình ảnh, tức là trên raster 2D thông thường.

• Tạo thanh màu cho phiên bản ScalarMappable * mappable * bằng cách sử dụng colorbar () phương thức và imshow () hình ảnh có thể lập bản đồ vô hướng.

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

Ví dụ

import numpy as np
from matplotlib import pyplot as plt
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True
data2D = np.random.random((50, 50))
im = plt.imshow(data2D, cmap="copper_r")
plt.colorbar(im)
plt.show()

Đầu ra