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

Làm thế nào để thêm một thanh màu cho một biểu đồ hist2d trong Matplotlib?

Để thêm thanh màu cho hist2d âm mưu, chúng ta có thể truyền một đối tượng có thể ánh xạ vô hướng vào đối số của phương thức colorbar ().

Các bước

  • Tạo các điểm dữ liệu x và y bằng cách sử dụng numpy.

  • Tạo một hình và một tập hợp các khe phụ bằng cách sử dụng subplots () phương pháp.

  • Tạo biểu đồ 2D bằng cách sử dụng hist2d () phương pháp.

  • Tạo một thanh màu cho một phiên bản có thể lập bản đồ vô hướng lịch sử.

  • Để 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, colors
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True
x = np.random.randn(100)
y = np.random.randn(100) + 5
fig, ax = plt.subplots()
hh = ax.hist2d(x, y, bins=40, norm=colors.LogNorm())
fig.colorbar(hh[3], ax=ax)
plt.show()

Đầu ra

Làm thế nào để thêm một thanh màu cho một biểu đồ hist2d trong Matplotlib?