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

วิธีเปลี่ยนสีสำหรับค่าเฉพาะใน Colormap ของ Matplotlib

หากต้องการกำหนดสีใหม่สำหรับค่าที่อยู่นอกช่วง (out-of-range) ใน colormap ของ Matplotlib สามารถทำตามขั้นตอนด้านล่างได้:

  • ดึงอินสแตนซ์ colormap โดยใช้เมธอด get_cmap() (ตัวอย่างใช้ colormap 'gray')
  • ตั้งค่าสีสำหรับค่าที่ต่ำกว่าช่วงเมื่อ norm.clip = False ด้วยเมธอด set_under()
  • แสดงข้อมูลเป็นภาพ 2D ด้วยเมธอด imshow()
  • แสดงกราฟด้วยเมธอด show()

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

import numpy as np
from matplotlib import pyplot as plt, cm

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

cmap = cm.get_cmap('gray')
cmap.set_under('red')
plt.imshow(np.arange(25).reshape(5, 5),
           interpolation='none',
           cmap=cmap,
           vmin=.001)
plt.show()

ผลลัพธ์

วิธีเปลี่ยนสีสำหรับค่าเฉพาะใน Colormap ของ Matplotlib