Matplotlib 顏色圖設置為黑色低於閾值

Question

我正在使用imshow到 plot 一個稀疏矩陣，並希望將 0 個條目塗成黑色。 我遵循了這個答案中給出的建議，但是我的 plot 仍然有 0 個條目的白色，這令人困惑，因為權重最高的條目是熱黃色。 任何幫助深表感謝。

這是我的代碼：

cmap1 = cm.get_cmap('inferno', 128)
cmap1.set_under(color='black')
im_plot = ax1.imshow(P_im,cmap=cmap1,norm=LogNorm(vmin=1e-30, vmax=np.max(P_im)+1e-15))
ax1.set_title("Title",size=10)

Answer 1

檢查此代碼：

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

fig, ax1 = plt.subplots(1, 1, figsize = (4, 4))

x = np.random.binomial(n = 1, p = 0.1, size = (20, 20))

cmap1 = cm.get_cmap('Greys_r', 2)
im_plot = ax1.imshow(x, cmap = cmap1)
ax1.set_title("Title", size = 10)

plt.show()

這給了我這張圖片：

我使用x = np.random.binomial(n = 1, p = 0.1, size = (20, 20))生成隨機稀疏矩陣，將其替換為您的數據。