Chỉ vẽ tam giác trên/duôi của heatmap trong Matplotlib

Để chỉ vẽ tam giác trên hoặc dưới của heatmap, ta dùng NumPy để tạo mảng mask 2D và chuyển thành hình ảnh.

Các bước thực hiện

• Thiết lập kích thước hình và khoảng đệm.
• Tạo dữ liệu ngẫu nhiên 5x5.
• Dùng numpy.tri() tạo mảng có 1 ở đường chéo và dưới, 0 ở nơi khác.
• Tạo masked array từ dữ liệu và mask.
• Hiển thị bằng imshow().
• Hiển thị bằng show().

Ví dụ

import numpy as np
from matplotlib import pyplot as plt

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

data = np.random.rand(5, 5)
mask = np.tri(data.shape[0], k=-1)
data = np.ma.array(data, mask=mask)

plt.imshow(data, interpolation="nearest", cmap='copper')
plt.show()

Kết quả