2011-07-16 82 views
11

關於堆棧溢出有很多matplotlib colorbar問題,但我無法理解它們以解決我的問題。如何添加Matplotlib Colorbar Ticks

如何在彩條上設置yticklabels?

下面是一些示例代碼:

from pylab import * 
from matplotlib.colors import LogNorm 
import matplotlib.pyplot as plt 

f = np.arange(0,101)     # frequency 
t = np.arange(11,245)    # time 
z = 20*np.sin(f**0.56)+22   # function 
z = np.reshape(z,(1,max(f.shape))) # reshape the function 
Z = z*np.ones((max(t.shape),1))  # make the single vector to a mxn matrix 
T, F = meshgrid(f,t) 
fig = plt.figure() 
ax = fig.add_subplot(111) 
plt.pcolor(F,T,Z, norm=LogNorm(vmin=z.min(),vmax=z.max())) 
plt.xlim((t.min(),t.max())) 
mn=int(np.floor(Z.min()))  # colorbar min value 
mx=int(np.ceil(Z.max()))   # colorbar max value 
md=(mx-mn)/2      # colorbar midpoint value 
cbar=plt.colorbar()    # the mystery step ??????????? 
cbar.set_yticklabels([mn,md,mx]) # add the labels 
plt.show() 
+0

我使用pydev的eclipse,有時我只是將代碼粘貼到python命令行中。我不完全確定你的建議,但我會研究,因爲這聽起來很有幫助。 – sequoia

+0

謝謝,很好的提示,看看未來有什麼方法可用。 – sequoia

回答

19

更新蜱和刻度標記:與沿杆5個蜱

cbar.set_ticks([mn,md,mx]) 
cbar.set_ticklabels([mn,md,mx]) 
+0

謝謝。這工作。 – sequoia

+0

爲什麼我的'mx'刻度是唯一不能在顏色條中顯示的?這怎麼會發生? – FaCoffee

5

工作示例(對於任何數值範圍)是:

m0=int(np.floor(field.min()))   # colorbar min value 
m4=int(np.ceil(field.max()))    # colorbar max value 
m1=int(1*(m4-m0)/4.0 + m0)    # colorbar mid value 1 
m2=int(2*(m4-m0)/4.0 + m0)    # colorbar mid value 2 
m3=int(3*(m4-m0)/4.0 + m0)    # colorbar mid value 3 
cbar.set_ticks([m0,m1,m2,m3,m4]) 
cbar.set_ticklabels([m0,m1,m2,m3,m4]) 
1

treenick回答讓我開始,但如果您的顏色條在0和1之間縮放,那麼該代碼將不會繪製蜱如果您fields不爲0之間縮放和1所以不是我用

m0=int(np.floor(field.min()))   # colorbar min value 
m4=int(np.ceil(field.max()))    # colorbar max value 
num_ticks = 10 
# to get ticks 
ticks = np.linspace(0, 1, num_ticks) 
# get labels 
labels = np.linspace(m0, m1, num_ticks) 

如果你想隔開標籤,你可以做Python列表索引像這樣: 假設跳過所有其他蜱

ticks = ticks[::2] 
labels = labels[::2]