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

การตั้งค่าขอบเขต Colorbar ของ Contour Plot ใน Matplotlib


เพื่อตั้งค่าขอบเขต (limits) ของ Colorbar ใน Contour Plot

  • ตั้งค่าขนาดรูปและ padding ระหว่าง subplots
  • สร้างข้อมูล x, y ด้วย numpy
  • คำนวณข้อมูลจาก x, y
  • สร้าง coordinate matrices ด้วย meshgrid
  • กำหนด vmin และ vmax สำหรับขอบเขต colorbar
  • พลอต contour ด้วย contourf()
  • สร้าง colorbar จาก ScalarMappable ในช่วง vmin-vmax
  • แสดงรูปด้วย show()

ตัวอย่าง

from matplotlib import pyplot as plt
import numpy as np
from matplotlib.cm import ScalarMappable

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

x = np.arange(20)
y = np.arange(20)
data = x[:, None] + y[None, :]
X, Y = np.meshgrid(x, y)
vmin = 0
vmax = 15

fig, ax = plt.subplots()
qcs = ax.contourf(X, Y, data, vmin=vmin, vmax=vmax)
fig.colorbar(
    ScalarMappable(norm=qcs.norm, cmap=qcs.cmap),
    ticks=range(vmin, vmax+5, 5)
)
plt.show()

ผลลัพธ์

การตั้งค่าขอบเขต Colorbar ของ Contour Plot ใน Matplotlib