Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Cách định nghĩa lại màu cho giá trị cụ thể trong colormap Matplotlib

Dùng set_under() trên colormap để đổi màu cho giá trị dưới vmin (khi norm.clip=False).

  • cmap = cm.get_cmap('gray').
  • cmap.set_under('red').
  • plt.imshow(data, interpolation='none', cmap=cmap, vmin=.001).
  • show().

Ví dụ

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()

Kết quả

Ảnh 5x5 grayscale, giá trị 0 (dưới vmin=0.001) hiển thị màu đỏ, các giá trị khác xám.