วิธีวาดเมทริกซ์ 2D พร้อม colorbar ใน Python ด้วย Matplotlib

แสดงเมทริกซ์ 2D เป็นภาพพร้อมแถบสี

ใช้ NumPy สร้างอาร์เรย์ 2 มิติ และนำไปใช้กับเมธอด imshow() พร้อมสร้าง colorbar

ขั้นตอน

1. สร้าง data2D ด้วย NumPy
2. ใช้ imshow() แสดงข้อมูลเป็นภาพบน grid 2 มิติ
3. สร้าง colorbar สำหรับ ScalarMappable instance ด้วยเมธอด colorbar() และอิมเมจจาก imshow()
4. แสดงภาพด้วย show()

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

import numpy as np
from matplotlib import pyplot as plt

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

data2D = np.random.random((50, 50))
im = plt.imshow(data2D, cmap="copper_r")
plt.colorbar(im)
plt.show()

ผลลัพธ์