Để thêm chú giải vào imshow () 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 dữ liệu ngẫu nhiên bằng cách sử dụng numpy.
- Khởi tạo bản đồ màu.
- Nhận các điểm dữ liệu duy nhất từ dữ liệu mẫu, bước 2.
- Vẽ từng màu với các nhãn và màu khác nhau để đặt vào chú giải.
- Đặt chú giải ở góc trên bên phải trong hộ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, cm plt.rcParams["figure.figsize"] = [7.50, 3.50] plt.rcParams["figure.autolayout"] = True data = np.random.rand(3, 3) cmap = cm.YlOrBr unique_data = np.unique(data) i = 0 for entry in unique_data: mycolor = cmap(entry * 255 / (max(unique_data) - min(unique_data))) plt.plot(0, 0, "-", color=mycolor, label="%d"%i) i += 1 plt.imshow(data, cmap=cmap) plt.legend(loc="upper right", bbox_to_anchor=(1.25, 1.0)) plt.show()
Đầu ra