matplotlib colorbar交替頂部底部標籤

Question

首先，這是作爲一個自我回答的問題，因爲我相信這在某些情況下會有幫助，例如，在this post作者試圖隱藏每隔一個標籤，以避免文本重疊，一個替代可能是交替標籤位置，以便保留所有標籤，避免重疊（如果沒有瘋狂的標籤數量）以及，這是這篇文章的目的是要解決的問題：

如何使matplotlib顏色條與交替的頂部和底部標籤？

Answer 1

跳轉到一個簡單的工作示例：

import numpy 
import matplotlib.pyplot as plt 

#------------------Get some data------------------ 
X = numpy.arange(100) 
Y = numpy.arange(100) 
Z = numpy.arange(100**2).reshape((100,100)) 

levels=numpy.arange(0,100**2,1000) 
ltop=levels[::2]   # labels appear on top 
lbot=levels[1:][::2]  # labels appear at bottom 

#-----------------------Plot----------------------- 
f = plt.figure() 
ax = f.gca() 

cf = ax.contourf(X,Y,Z,100) 
cbar=plt.colorbar(cf,orientation='horizontal',ticks=lbot,drawedges=True) 

vmin=cbar.norm.vmin 
vmax=cbar.norm.vmax 

#-------------Print bottom tick labels------------- 
cbar.ax.set_xticklabels(lbot) 

#--------------Print top tick labels-------------- 
for ii in ltop: 
    cbar.ax.text((ii-vmin)/(vmax-vmin), 1.5, str(ii), transform=cbar.ax.transAxes, va='bottom', ha='center') 

plt.show(block=False) 

基本上底部標籤使用默認方法cbar.ax.set_xticklabels(lbot)作圖。對於頂級標籤，我使用cbar.ax.text()手動添加了它們。

情節看起來像這樣：

編輯：重要更新我的回答：

當彩條具有延伸/溢出，三角形被用在相關的端部，以指示值溢出。在這種情況下，頂行標記標籤需要進行一些調整才能正確對齊顏色條部分。

默認情況下，三角形大小是彩條軸的5％，這用於獲得正確的偏移和縮放以對齊標籤。

查看下面的示例，該示例在兩端都有擴展。用我以前的方法，結果是這樣的：

2.在頂部行結束數與三角形的頂端對齊。如果只有一端延伸並且輪廓級數很大（> = 10左右），則不對中會變得更糟。

校正後的情節：

這是產生正確的陰謀代碼：

import numpy 
import matplotlib.pyplot as plt 

#------------------Get some data------------------ 
X = numpy.linspace(-1,1,100) 
Y = numpy.linspace(-1,1,100) 
X,Y=numpy.meshgrid(X,Y) 
Z=numpy.sin(X**2) 

levels=numpy.linspace(-0.8,0.8,9) 

ltop=levels[::2]   # labels appear on top 
lbot=levels[1:][::2]  # labels appear at bottom 

#-----------------------Plot----------------------- 
f = plt.figure() 
ax = f.gca() 

cf = ax.contourf(X,Y,Z,levels,extend='both') 
cbar=plt.colorbar(cf,orientation='horizontal',ticks=lbot,drawedges=True) 

#------------Compute top tick label locations------------ 
vmin=cbar.norm.vmin 
vmax=cbar.norm.vmax 

if cbar.extend=='min': 
    shift_l=0.05 
    scaling=0.95 
elif cbar.extend=='max': 
    shift_l=0. 
    scaling=0.95 
elif cbar.extend=='both': 
    shift_l=0.05 
    scaling=0.9 
else: 
    shift_l=0. 
    scaling=1.0 

#-------------Print bottom tick labels------------- 
cbar.ax.set_xticklabels(lbot) 

#--------------Print top tick labels-------------- 
for ii in ltop: 
    cbar.ax.text(shift_l + scaling*(ii-vmin)/(vmax-vmin), 
     1.5, str(ii), transform=cbar.ax.transAxes, 
     va='bottom', ha='center') 

plt.show(block=False) 

Answer 2

你可以添加一個雙軸對象和每個奇數刻度設置那裏，而即使設置每在原始軸上滴答滴答。

import numpy as np 
import matplotlib.pyplot as plt 

# Make the plot 
fig, ax = plt.subplots(1,1, figsize=(5,5)) 
fig.subplots_adjust(bottom=0.2) 
## Trick to have the colorbar of the same size as the plot 
box = ax.get_position() 
cax = fig.add_axes([box.xmin, box.ymin - 0.1, box.width, 0.03]) 
m = ax.matshow(np.random.random(100).reshape(10,10), aspect="auto") # Don't forget auto or the size of the heatmap will change. 
cb = plt.colorbar(m, cax=cax, orientation="horizontal") 

# Add twin axes 
cax2 = cax.twiny() 

# get current positions and values of the ticks. 
# OR you can skip this part and set your own ticks instead. 
xt = cax.get_xticks() 
xtl = [i.get_text() for i in cax.get_xticklabels()] 

# set odd ticks on top (twin axe) 
cax2.set_xticks(xt[1::2]) 
cax2.set_xticklabels(xtl[1::2]) 
# set even ticks on original axes (note the different object : cb != cax) 
cb.set_ticks(xt[::2]) 
cb.set_ticklabels(xtl[::2]) 

HTH