Làm thế nào để vẽ biểu đồ dữ liệu vào imshow () với bản đồ màu tùy chỉnh trong Matplotlib?

Để vẽ biểu đồ dữ liệu vào imshow () với bản đồ màu tùy chỉnh trong matplotlib, chúng ta có thể thực hiện các bước sau đây−

• Đặt kích thước hình và điều chỉnh phần đệm giữa và xung quanh các ô phụ.
• Tạo các điểm dữ liệu ngẫu nhiên bằng cách sử dụng numpy.
• Tạo bản đồ màu đối tượng từ danh sách màu sắc.
• Hiển thị dữ liệu dưới dạng hình ảnh, tức là trên raster 2D thông thường
• Để 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
from matplotlib.colors import ListedColormap
import numpy as np
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True
data = np.random.rand(5, 5)
cmap = ListedColormap(['r', 'g', 'b'])
plt.imshow(data, cmap=cmap)
plt.show()

Đầu ra