Cách vẽ ma trận 2D với colorbar trong Python Matplotlib

Dùng imshow() để vẽ ma trận 2D và colorbar() để thêm thanh màu.

• Tạo data2D = np.random.random((50, 50)).
• im = plt.imshow(data2D, cmap="copper_r").
• plt.colorbar(im).
• show().

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

Kết quả

Heatmap 50x50 với colormap copper_r và colorbar bên phải.