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

Làm thế nào để hiển thị nhiều thanh màu trong Matplotlib?

Để hiển thị nhiều thanh màu trong matplotlib, chúng ta có thể thực hiện các bước sau đây−

  • Đặ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 một hình và một tập hợp các tình huống phụ.
  • Khởi tạo biến N cho số lượng dữ liệu mẫu.
  • Tạo data1 ngẫu nhiên sử dụng numpy.
  • Hiển thị dữ liệu dưới dạng hình ảnh, tức là trên raster 2D thông thường, với data1 .
  • Thêm thanh màu vào một âm mưu.
  • Lặp lại các bước 4, 5 và 6 với các bộ dữ liệu và trục khác nhau.
  • Để hiển thị hình này, hãy sử dụng show () phương pháp.

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()

Đầu ra

Làm thế nào để hiển thị nhiều thanh màu trong Matplotlib?