matplotlib colorbar чередующиеся верхние нижние метки

Question

Прежде всего, это было задано как вопрос с самостоятельным ответом, потому что я считаю, что это будет полезно в определенных ситуациях, например, в этом посте автор попытался скрыть все остальные ярлыки, чтобы избежать наложения текстов, одна альтернатива может состоять в чередовании ярлыка Положение так, чтобы сохранить все метки и избежать наложения (если не существует сумасшедшего количества меток), что и является целью этого поста:

Как сделать matplotlib colorbar с чередующимися метками сверху и снизу?

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])

НТН