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:
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:
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 is a backend software engineer at Matrice.ai.
LinkedIn