Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Cách hiển thị nhiều thanh màu (colorbars) trong Matplotlib

Để hiển thị nhiều colorbars trong matplotlib, thực hiện các bước sau:

  • Thiết lập kích thước hình và điều chỉnh khoảng đệm giữa các subplot.
  • Tạo figure và tập hợp subplots.
  • Khởi tạo biến N cho số lượng dữ liệu mẫu.
  • Tạo dữ liệu ngẫu nhiên data1 bằng numpy.
  • Hiển thị dữ liệu dưới dạng ảnh 2D với imshow().
  • Thêm colorbar cho subplot.
  • Lặp lại với các dữ liệu và trục khác nhau.
  • Sử dụng show() để hiển thị.

Ví dụ

from matplotlib import pyplot as plt
import numpy as np
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True
fig, (ax1, ax2) = plt.subplots(nrows=2, ncols=1)
N = 10
data1 = np.random.rand(N, N)
im1 = ax1.imshow(data1, cmap='hot')
plt.colorbar(im1, cax=ax1)
data2 = np.random.rand(N, N)
im2 = ax2.imshow(data2, cmap='plasma')
plt.colorbar(im2, cax=ax2)
plt.show()

Kết quả

Hiển thị hình ảnh với hai colorbars riêng biệt cho từng subplot.