分配特定值Matplotlib圖

Question

我有一個numpy數組中的兩個值。 0,1。我想確保1在繪製數組時總是黑色的，而0總是灰色的。

我該如何在matplotlib中做到這一點？

謝謝

Answer 1

假設要繪製的圖像圖（plt.imshow()），你可以選擇喜歡它有白色作爲最低和黑色作爲最高顏色「gray_r」一個顏色映射，灰色將是在中間。如果現在將圖像圖標歸一化爲-1和1之間的值，則數組的0值將對應於色彩圖的中間（灰色），1將對應於色彩圖的上端（黑色）。

import matplotlib.pyplot as plt 
import numpy as np; np.random.seed(0) 

a = np.random.randint(0,2, size=(12,25)) 

plt.imshow(a, cmap="gray_r", vmin=-1, vmax=1) 

plt.show()