Computer >> Máy Tính >  >> Lập trình >> Python

Làm thế nào để gắn nhãn cho các thanh màu trong biểu đồ imshow của Matplolib?

Để cung cấp matplotlib imshow () vẽ các thanh màu cho một nhãn, 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 ô con.

  • Tạo điểm dữ liệu 5 × 5 bằng Numpy.

  • Sử dụng imshow () phương pháp hiển thị dữ liệu dưới dạng hình ảnh, tức là trên raster 2D thông thường.

  • Tạo thanh màu cho ScalarMappable ví dụ, tôi.

  • Đặt nhãn thanh màu bằng set_label () phương pháp.

  • Để hiển thị hình này, hãy sử dụng show () phương pháp.

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)

im = plt.imshow(data, cmap="copper")

cbar = plt.colorbar(im)
cbar.set_label("Colorbar")

plt.show()

Đầu ra

Làm thế nào để gắn nhãn cho các thanh màu trong biểu đồ imshow của Matplolib?