การใช้ Colorbar กับ Hist2D ใน Matplotlib Pyplot

ขั้นตอนการทำงาน

• กำหนดขนาดรูปภาพและปรับระยะห่างรอบๆ subplots

• กำหนดตัวแปร N สำหรับจำนวนข้อมูลตัวอย่าง

• สร้างข้อมูล x และ y ด้วย NumPy

• สร้าง Figure และ Subplots

• สร้าง 2D Histogram ด้วย hist2d()

• สร้าง Colorbar จาก Scalar Mappable ของ hist2d

• แสดงรูปภาพด้วยเมธอด show()

ตัวอย่างโค้ด

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

N = 1000
x = np.random.rand(N)
y = np.random.rand(N)

fig, ax = plt.subplots()
hh = ax.hist2d(x, y, bins=40, norm=LogNorm())
fig.colorbar(hh[3], ax=ax)
plt.show()

ผลลัพธ์

2D Histogram (Heatmap) แสดงความหนาแน่นของจุดข้อมูล 1000 จุด ใช้ LogNorm สำหรับสเกลสี พร้อม Colorbar แสดงจำนวนจุดในแต่ละ bin

คำอธิบาย: hh[3] คือ QuadMesh object (ScalarMappable) ที่สร้างจาก hist2d ใช้สร้าง Colorbar ได้โดยตรง