光譜圖 - 美白出特定區域

Question

我在C++中生成光譜圖，並使用matplotlib繪製/顯示此圖。

這很好，但是，我正在分割譜圖的一部分，並且想要確定關鍵興趣點。式的工作原理類似於這樣：

if(block >= threshold) then 
    spectrogram = block 
else: 
    spectrogram = "0" 

在本質上，如果該頻譜內的塊是上方的閾值，則該塊的數據在特定的位置被捕獲。但是，如果不是那麼我們只是讓塊包含「0」。問題是，當我繪製這一點，它不會在值爲「0」的區域顯示「白色」：

import numpy as np 
import matplotlib.pylab as pl; 

data = np.loadtxt('spec.txt') 

hm = pl.imshow(10*np.log10(data.T),interpolation='nearest',origin='lower',aspect='auto') 
pl.show() 

下面是結果：

是否有設置的方式，使不感興趣的區域爲「白色」，只顯示感興趣的區域？你想白

data = np.loadtxt("spec.txt") 
data = np.ma.masked_where(data == 0, data) 
pl.imshow(data, interpolation='nearest', ...) 

編輯

Answer 1

嘗試掩蓋方面：

下面是一個例子：

import numpy as np 
import matplotlib.pyplot as plt 

data = np.random.random_integers(-1, 1, (10, 10)) 
masked_data = np.ma.masked_where(data == 0, data) 

plt.imshow(data, interpolation="nearest") 
plt.show() 

plt.imshow(masked_data, interpolation="nearest") 
plt.show() 

第二個編輯：

要更改配色方案，使用顏色表

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

data = np.random.rand(10, 10) 

cmap = matplotlib.cm.binary 
plt.imshow(data, interpolation="nearest", cmap=cmap) 
plt.show() 

退房here對於許多色彩映射的列表。