Matplotlib: задний фон (оси + метки) цвет [или, системы координат фигур/осей]

У меня есть фигура, содержащая подстроки 3x2, и я хотел бы установить фоновый цвет в средней паре подзаголовков, чтобы сделать более ясным, какие метки осей принадлежат к какой подзаголовке.

Настройка facecolor при построении подзаголовки изменяет только цвет области, определяемой осями; метки меток и оси по-прежнему нарисованы на figure.patch. Предполагая, что нет простого способа сделать это, я мог бы добавить прямоугольный патч за соответствующими экземплярами в figure.axes.

После небольшого эксперимента, похоже, что figure.axes[x].get_position() возвращает Axes coördinates (нормализованные coördinates [0.0-1.0]), но Rectangle(), похоже, хочет отображать coördinates (пиксели). Этот код более или менее работает (ED: в интерактивном режиме, но при выводе на png (с использованием средства рендеринга Agg) позиционирование прямоугольника полностью отключается):

import matplotlib.pyplot as plt
import matplotlib

f = plt.figure()
plt.subplot( 121 )
plt.title( 'model' )
plt.plot( range(5), range(5) )
plt.xlabel( 'x axis' )
plt.ylabel( 'left graph' )
plt.subplot( 122 )
plt.title( 'residuals' )
plt.plot( range(5), range(5) )
plt.xlabel( 'x axis' )
plt.ylabel( 'right graph' )
plt.tight_layout(pad=4)

bb = f.axes[0].get_position().transformed( f.transFigure ).get_points()
bb_pad = (bb[1] - bb[0])*[.20, .10]
bb_offs = bb_pad * [-.25, -.20]
r = matplotlib.patches.Rectangle( bb[0]-bb_pad+bb_offs, *(bb[1] - bb[0] + 2*bb_pad),
                                  zorder=-10, facecolor='0.85', edgecolor='none' )
f.patches.extend( [r] )

но кажется очень хакерским, и мне кажется, что я пропустил что-то важное. Может кто-нибудь объяснить, что, есть ли более простой/лучший способ сделать это, и если да, то что это такое?

Поскольку мне действительно нужно писать в файл, у меня сейчас нет решения.

Ответ 1

Просто используйте transform kwarg для Rectangle, и вы можете использовать любую систему координат, которая вам нужна.

В качестве простого примера:

import matplotlib.pyplot as plt
from matplotlib.patches import Rectangle

fig, axes = plt.subplots(3, 2)

rect = Rectangle((0.08, 0.35), 0.85, 0.28, facecolor='yellow', edgecolor='none',
                 transform=fig.transFigure, zorder=-1)
fig.patches.append(rect)
plt.show()

Однако, если вы хотите сделать что-то более надежно и рассчитать степень осей, вы можете сделать что-то вроде этого:

import matplotlib.pyplot as plt
from matplotlib.transforms import Bbox
from matplotlib.patches import Rectangle

def full_extent(ax, pad=0.0):
    """Get the full extent of an axes, including axes labels, tick labels, and
    titles."""
    # For text objects, we need to draw the figure first, otherwise the extents
    # are undefined.
    ax.figure.canvas.draw()
    items = ax.get_xticklabels() + ax.get_yticklabels() 
#    items += [ax, ax.title, ax.xaxis.label, ax.yaxis.label]
    items += [ax, ax.title]
    bbox = Bbox.union([item.get_window_extent() for item in items])
    return bbox.expanded(1.0 + pad, 1.0 + pad)


fig, axes = plt.subplots(3,2)

extent = Bbox.union([full_extent(ax) for ax in axes[1,:]])

# It best to transform this back into figure coordinates. Otherwise, it won't
# behave correctly when the size of the plot is changed.
extent = extent.transformed(fig.transFigure.inverted())

# We can now make the rectangle in figure coords using the "transform" kwarg.
rect = Rectangle([extent.xmin, extent.ymin], extent.width, extent.height,
                 facecolor='yellow', edgecolor='none', zorder=-1, 
                 transform=fig.transFigure)
fig.patches.append(rect)

plt.show()