Làm thế nào để hiển thị giá trị ma trận và bản đồ màu trong Matplotlib?

Để hiển thị giá trị ma trận và bản đồ màu trong Matplotlib, chúng ta có thể thực hiện các bước sau -

• Đặ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 một hình và một tập hợp các tình huống phụ.
• Khởi tạo các giá trị tối đa và giá trị nhỏ nhất cho ma trận.
• Vẽ đồ thị các giá trị của ma trận hoặc mảng 2D dưới dạng hình ảnh được mã hóa màu.
• Lặp lại từng ô của hình ảnh mã màu và đặt giá trị ở giữa.
• Để hiển thị hình này, hãy sử dụng show () phương pháp.

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

Đầu ra