2017-04-21 39 views
1

我目前正在試圖用matplotlib形象化一些數據集,但似乎有一些顏色範圍的問題。使用matplotlibs製作圖的固定colorange

我目前做我的繪圖方式是以下幾點:密謀這種方式是,彩條自動調整到數據集,這是不想要

def make_plot_store_data(name,interweaved,static): 
    Y = np.array(range(0,static.shape[1])) 
    X = np.array(range(0,static.shape[0])) 
    X,Y = np.meshgrid(X, Y) 

    plt.pcolormesh(X,Y,np.log10(static.T),cmap=cm.jet) 
    plt.xlabel('Frames') 
    plt.ylabel('Frequency(Hz)') 
    plt.title('Power spectrum of ' + name) 
    plt.colorbar() 
    plt.savefig(plot+"/"+name+"_plot_static_conv.png") 
    plt.close() 

問題。

我希望有一個恆定彩條/固定顏色條使得i避免表示這樣的:

enter image description here

在其暗紅色=> 18

在這裏它是:

enter image description here

變黑=> 2.5

我正在嘗試模式識別,我認爲這有點搞亂我的結果。

回答

0

可以使用VMINVMAX設定的最小和最大色彩層次:

import numpy as np 
import matplotlib.pyplot as plt 

x = np.arange(0,100,0.2) 
y = np.arange(0,40,0.2) 
X, Y = np.meshgrid(x,y) 
Z = np.random.random((y.shape[0],x.shape[0]))*np.sin(0.01*X*Y) 


plt.pcolormesh(X,Y,Z, cmap=plt.cm.jet, vmin=0, vmax=1) 
plt.colorbar() 
plt.show() 

enter image description here