Thêm nhãn cho colorbar của imshow trong Matplotlib

Để thêm nhãn (label) cho colorbar của đồ thị imshow() trong Matplotlib:

• Thiết lập kích thước figure và padding
• Tạo dữ liệu 5x5 bằng Numpy
• Hiển thị dữ liệu như ảnh bằng imshow()
• Tạo colorbar cho ScalarMappable instance
• Đặt nhãn colorbar bằng set_label()
• Hiển thị bằng show()

Ví dụ mã nguồn

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)
im = plt.imshow(data, cmap="copper")
cbar = plt.colorbar(im)
cbar.set_label("Colorbar")
plt.show()

Kết quả

Ảnh heatmap với colorbar bên cạnh có nhãn "Colorbar".

Mở rộng: Có thể tùy chỉnh thêm: cbar.set_label('Label', rotation=270, labelpad=15) để xoay nhãn và điều chỉnh khoảng cách.