ขั้นตอนการทำงาน
- กำหนดขนาดรูปภาพและปรับระยะห่างระหว่างและรอบๆ subplots
- สร้างข้อมูล x, y และ z ด้วย numpy
- ใช้เมธอด imshow() แสดงข้อมูลเป็นภาพบนกริด 2D
- สร้าง ticks ด้วย numpy ในช่วงค่าต่ำสุดและสูงสุดของ z
- สร้าง colorbar สำหรับ ScalarMappable instance พร้อมกำหนด ticks=ticks
- ใช้เมธอด show() เพื่อแสดงรูปภาพ
ตัวอย่างโค้ด
import numpy as np import matplotlib.pyplot as plt plt.rcParams["figure.figsize"] = [7.50, 3.50] plt.rcParams["figure.autolayout"] = True x, y = np.mgrid[-1:1:100j, -1:1:100j] z = (x + y) * np.exp(-5.0 * (x ** 2 + y ** 2)) plt.imshow(z, extent=[-1, 1, -1, 1]) ticks = np.linspace(z.min(), z.max(), 5, endpoint=True) cb = plt.colorbar(ticks=ticks) plt.show()
ผลลัพธ์
