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

Cách vẽ colorbar của pcolormesh ở subplot riêng trong Matplotlib


Để đặt colorbar cho từng pcolormesh ở subplot riêng biệt, bạn tạo lưới subplot và gọi fig.colorbar(pcm, ax=ax) cho từng axes.

  • Thiết lập kích thước hình vẽ và khoảng đệm.
  • Tạo figure và lưới subplot 2x2.
  • Định nghĩa danh sách colormap.
  • Lặp qua từng axes, vẽ pcolormesh với dữ liệu ngẫu nhiên.
  • Tạo colorbar cho mỗi pcolormesh gắn với axes tương ứng.
  • Hiển thị hình vẽ.

Ví dụ minh họa

import matplotlib.pyplot as plt
import numpy as np

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

fig, axs = plt.subplots(2, 2)
cm = ['plasma', 'copper']

for col in range(2):
    for row in range(2):
        ax = axs[row, col]
        pcm = ax.pcolormesh(np.random.random((20, 20)) * (col + 1), cmap=cm[col])
        fig.colorbar(pcm, ax=ax)

plt.show()

Kết quả đầu ra

Cách vẽ colorbar của pcolormesh ở subplot riêng trong Matplotlib