Matplotlib: Как добавить текст на рисунок за пределами осей?

Краткий обзор Matplotlib и его структуры рисунка

Matplotlib – это мощная библиотека Python для визуализации данных, позволяющая создавать широкий спектр графиков, от простых диаграмм до сложных анимаций. Основная структура рисунка Matplotlib включает в себя Figure (контейнер для всего рисунка) и Axes (области, где отображаются сами данные). Figure может содержать несколько Axes, и именно на Axes происходит рисование графиков. Однако, иногда требуется разместить текст, который относится ко всему рисунку, а не к отдельным осям, например, заголовок, подпись или водяной знак.

Проблема размещения текста за пределами осей и ее решения

Размещение текста за пределами Axes представляет собой определенную задачу, поскольку стандартные методы работы с текстом в Matplotlib ориентированы на Axes. Необходимо использовать методы, которые позволяют оперировать координатами всего рисунка Figure, а не координатными осями. Существует несколько подходов к решению этой задачи, каждый из которых имеет свои преимущества и недостатки. Мы рассмотрим figtext() и использование transforms для точного позиционирования.

Основные способы добавления текста за пределами осей

Использование figtext() для добавления текста на весь рисунок

Функция figtext() является одним из наиболее простых способов добавления текста на Figure. Она принимает координаты x и y в системе координат Figure (от 0 до 1), где (0, 0) – нижний левый угол, а (1, 1) – верхний правый угол.

import matplotlib.pyplot as plt

def add_figure_text(text: str, x: float, y: float, fontsize: int = 12, color: str = 'black') -> None:
    """Adds text to a Matplotlib figure outside of the axes.

    Args:
        text: The text to add.
        x: The x-coordinate of the text (0 to 1).
        y: The y-coordinate of the text (0 to 1).
        fontsize: The font size of the text.
        color: The color of the text.
    """
    fig = plt.figure()
    fig.text(x, y, text, fontsize=fontsize, color=color)
    plt.show()

# Example usage
add_figure_text("Figure Title", 0.5, 0.95, fontsize=16, color='blue')

Указание координат в системе координат рисунка (Figure coordinates)

Как упоминалось выше, координаты x и y в figtext() задаются в диапазоне от 0 до 1. Это позволяет легко размещать текст относительно размеров всего рисунка, независимо от размеров и расположения осей Axes.

Настройка шрифта, размера и цвета текста

Функция figtext() также позволяет настраивать различные параметры текста, такие как шрифт (fontfamily, fontweight), размер (fontsize), цвет (color), выравнивание (ha, va) и другие. Это дает полный контроль над внешним видом текста.

import matplotlib.pyplot as plt

def add_styled_figure_text(text: str, x: float, y: float, fontsize: int = 12, color: str = 'black', ha: str = 'center', va: str = 'center') -> None:
    """Adds styled text to a Matplotlib figure outside of the axes.

    Args:
        text: The text to add.
        x: The x-coordinate of the text (0 to 1).
        y: The y-coordinate of the text (0 to 1).
        fontsize: The font size of the text.
        color: The color of the text.
        ha: Horizontal alignment ('center', 'left', 'right').
        va: Vertical alignment ('center', 'top', 'bottom', 'baseline', 'center_baseline').
    """
    fig = plt.figure()
    fig.text(x, y, text, fontsize=fontsize, color=color, ha=ha, va=va)
    plt.show()

# Example usage
add_styled_figure_text("Centered Text", 0.5, 0.5, fontsize=14, color='green', ha='center', va='center')

Продвинутые методы позиционирования текста

Использование transforms для точного размещения текста

transforms в Matplotlib позволяют преобразовывать координаты из одной системы в другую. Для размещения текста за пределами осей можно использовать Figure transform, который позволяет задавать координаты относительно всего рисунка.

import matplotlib.pyplot as plt
import matplotlib.transforms as transforms

def add_text_with_transform(text: str, x: float, y: float, fontsize: int = 12, color: str = 'black') -> None:
    """Adds text to a Matplotlib figure using transforms for precise positioning.

    Args:
        text: The text to add.
        x: The x-coordinate of the text (in figure coordinates).
        y: The y-coordinate of the text (in figure coordinates).
        fontsize: The font size of the text.
        color: The color of the text.
    """
    fig = plt.figure()
    ax = fig.add_subplot(111)
    # The following line ensures that the plot is displayed even if there is no data.
    ax.plot([0, 1], [0, 1], visible=False)

    transform = fig.transFigure.inverted()
    text_x, text_y = transform.transform((x * fig.get_figwidth(), y * fig.get_figheight()))

    fig.text(text_x, text_y, text, fontsize=fontsize, color=color)
    plt.show()

# Example usage
add_text_with_transform("Transformed Text", 0.1, 0.1, fontsize=10, color='purple')

Размещение текста относительно границ рисунка

Используя transforms, можно размещать текст относительно определенных границ рисунка, например, левого верхнего угла или правого нижнего угла. Это полезно при создании водяных знаков или логотипов.

Создание пользовательских функций для позиционирования

Для упрощения процесса добавления текста можно создать пользовательские функции, которые принимают, например, положение текста в виде строки («top left», «bottom right», «center») и автоматически рассчитывают координаты на основе размеров рисунка.

Примеры использования и распространенные ошибки

Пример: Добавление заголовка и подписи к рисунку за пределами осей

import matplotlib.pyplot as plt

def add_title_and_caption(title: str, caption: str) -> None:
    """Adds a title and caption to a Matplotlib figure outside of the axes.

    Args:
        title: The title of the figure.
        caption: The caption for the figure.
    """
    fig = plt.figure(figsize=(8, 6))
    ax = fig.add_subplot(111)
    ax.plot([1, 2, 3, 4], [5, 6, 7, 8])

    fig.text(0.5, 0.95, title, ha='center', fontsize=16)
    fig.text(0.5, 0.05, caption, ha='center', fontsize=10, color='gray')

    plt.show()

# Example usage
add_title_and_caption("Sales Trend", "Source: Internal Data")

Пример: Размещение логотипа или водяного знака

Для добавления логотипа или водяного знака можно использовать matplotlib.image.imread для загрузки изображения и fig.figimage для его размещения.

Распространенные ошибки и способы их исправления (обрезание текста, неправильные координаты)

• Обрезание текста: Если текст обрезается по краям рисунка, убедитесь, что координаты находятся в пределах диапазона [0, 1] и достаточно места для текста. Также можно использовать bbox_inches='tight' при сохранении рисунка.
• Неправильные координаты: Тщательно проверяйте координаты, особенно при использовании transforms. Помните, что координаты задаются в системе координат Figure.
• Наложение текста на оси: Убедитесь, что текст не перекрывает графики и оси, правильно выбрав координаты.

Заключение

Краткое повторение рассмотренных методов

Мы рассмотрели несколько способов добавления текста за пределами осей в Matplotlib: использование figtext(), настройку шрифта, размера и цвета текста, использование transforms для точного размещения текста. Каждый метод имеет свои преимущества и недостатки, и выбор зависит от конкретной задачи.

Дополнительные ресурсы и ссылки для дальнейшего изучения Matplotlib

• Официальная документация Matplotlib
• Галерея примеров Matplotlib
• Руководства по Matplotlib