Computer >> บทช่วยสอนคอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

วิธีเพิ่มเส้นทิก (Ticks) ให้กับ Colorbar ใน Matplotlib


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

  • กำหนดขนาดรูปภาพและปรับระยะห่างระหว่างและรอบๆ 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()

ผลลัพธ์

วิธีเพิ่มเส้นทิก (Ticks) ให้กับ Colorbar ใน Matplotlib