Especificar a Posição da Legenda em Coordenadas Gráficas em Matplotlib

Suraj Joshi 6 fevereiro 2021
Especificar a Posição da Legenda em Coordenadas Gráficas em Matplotlib

Podemos especificar a posição de uma legenda num gráfico, definindo o valor do parâmetro loc.

Exemplos: Especificar a posição da legenda num gráfico de coordenadas Matplotlib

import matplotlib.pyplot as plt

x = [1, 2, 3]
y1 = [0.5 * i + 1 for i in x]
y2 = [2 * i + 1 for i in x]

fig, ax = plt.subplots(2, 1)

ax[0].plot(x, y1, "b-", label="0.5x+1")
ax[0].plot(x, y2, "g-", label="2x+1")
ax[0].legend(loc="best")

ax[1].plot(x, y1, "b-", label="0.5x+1")
ax[1].plot(x, y2, "g-", label="2x+1")
ax[1].legend(loc="lower left")

plt.tight_layout()
plt.show()

Resultado:

Especificar a posição da legenda nas coordenadas gráficas Matplotlib

Aqui, temos uma figura com dois subquadrantes. O subplot mais alto tem loc definido para best; isto coloca a lenda na melhor posição da figura, onde não temos qualquer conteúdo.

Uma coisa semelhante acontece na segunda subparcela em baixo: a legenda é colocada na posição inferior esquerda da caixa delimitadora, cobrindo todos os eixos. A caixa delimitadora é especificada pelo parâmetro bbox_to_anchor, cujo valor por defeito é (0,0,1,1).

O parâmetro loc pode assumir qualquer um dos seguintes valores:

best
upper right
upper left
lower left
lower right
right
center left
center right
lower center
upper center
center

Estas posições representam a posição da legenda, em relação à caixa delimitadora especificada pelo parâmetro bbox_to_anchor.

Da mesma forma, podemos colocar a legenda em qualquer posição da figura alterando o valor do parâmetro bbox_to_anchor. O parâmetro bbox_to_anchor toma um tuple, representando a coordenada, onde será colocado o canto especificado pelo parâmetro loc.

import matplotlib.pyplot as plt

x = [1, 2, 3]
y1 = [0.5 * i + 1 for i in x]
y2 = [2 * i + 1 for i in x]

fig, ax = plt.subplots(2, 1)

ax[0].plot(x, y1, "b-", label="Plot-1")
ax[0].plot(x, y2, "g-", label="Plot-2")
ax[0].legend(loc="upper left", bbox_to_anchor=(0.9, 0.75))
ax[0].scatter((0.9), (0.75), s=70, c="red", transform=ax[0].transAxes)

ax[1].plot(x, y1, "b-", label="Plot-1")
ax[1].plot(x, y2, "g-", label="Plot-2")
ax[1].legend(loc="center right", bbox_to_anchor=(0.6, 0.4))
ax[1].scatter((0.6), (0.4), s=70, c="red", transform=ax[1].transAxes)

plt.tight_layout()
plt.show()

Resultado:

Definir bbox_to_anchor para especificar a posição da legenda nas coordenadas do gráfico Matplotlib

Na subparcela no topo, definimos o bbox_to_anchor a (0.9,0.75), que é marcado pelo ponto vermelho na subparcela. O valor do parâmetro loc "upper left" representa o canto superior esquerdo da legenda, que é colocado no ponto vermelho.

A subparcela em baixo, o bbox_to_anchor está definido como (0.6,0.4), marcado pelo ponto vermelho nos eixos inferiores da figura. O valor de loc está definido para "center right"; assim, a legenda dos eixos inferiores tem o seu canto centro direito no ponto vermelho.

Suraj Joshi avatar Suraj Joshi avatar

Suraj Joshi is a backend software engineer at Matrice.ai.

LinkedIn

Artigo relacionado - Matplotlib Legend