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

Sử dụng colorbar với hist2d trong matplotlib.pyplot

Để sử dụng colorbar với hist2d trong matplotlib.pyplot, bạn có thể thực hiện theo các bước sau:

Các bước thực hiện

  • Thiết lập kích thước hình vẽ và điều chỉnh khoảng cách giữa các subplot.
  • Khởi tạo biến N cho số lượng dữ liệu mẫu.
  • Tạo điểm dữ liệu x và y bằng NumPy.
  • Tạo figure và tập hợp subplot bằng subplots().
  • Vẽ biểu đồ histogram 2D bằng hist2d().
  • Tạo colorbar cho instance scalar mappable của hist2d.
  • Sử dụng phương thức show() để hiển thị hình vẽ.

Ví dụ minh họa

from matplotlib.colors import LogNorm
import matplotlib.pyplot as plt
import numpy as np

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

# Số lượng dữ liệu mẫu
N = 1000

# Tạo điểm dữ liệu x, y
x = np.random.rand(N)
y = np.random.rand(N)

fig, ax = plt.subplots()

# Histogram 2D với chuẩn hóa Log
hh = ax.hist2d(x, y, bins=40, norm=LogNorm())

# Tạo colorbar
fig.colorbar(hh[3], ax=ax)

plt.show()

Kết quả đầu ra

Biểu đồ histogram 2D với thang màu (colorbar) bên cạnh thể hiện mật độ điểm dữ liệu theo thang logarit.