Cách hiển thị giá trị ma trận và colormap trong Matplotlib?

Để hiển thị giá trị ma trận và colormap trong 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.
• Tạo một figure và một tập subplots.
• Khởi tạo giá trị max và min cho ma trận.
• Vẽ giá trị của ma trận 2D hoặc mảng dưới dạng hình ảnh mã hóa màu.
• Lặp qua từng ô của hình ảnh mã hóa màu và đặt giá trị ở trung tâm.
• Để hiển thị figure, dùng phương thức show().

Ví dụ

import numpy as np
import matplotlib.pyplot as plt

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

fig, ax = plt.subplots()
min_val, max_val = 0, 5
matrix = np.random.randint(0, 5, size=(max_val, max_val))

ax.matshow(matrix, cmap='ocean')

for i in range(max_val):
    for j in range(max_val):
        c = matrix[j, i]
        ax.text(i, j, str(c), va='center', ha='center')

plt.show()

Kết quả