Przenoszenie i zmiana rozmiaru legend-box w matplotlib

Question

Tworzę działki przy użyciu Matplotlib, które zapisuję jako SVG, eksportuję do .pdf + .pdf_tex za pomocą Inkscape i dołączam plik .pdf_tex do dokumentu LaTeX.

To oznacza, że ​​mogę wejść lateksu polecenia w tytuły, legendy itp, dając obraz jak ten

co czyni tak, gdy używam go w moim dokumencie LaTeX. Zauważ, że czcionka dla liczb na osi zmienić, a lateks kodu w legendzie jest kompilowany:

kod na działce (jak eksport do SVG nie pokazano tutaj, ale można wykazać na życzenie):

import numpy as np 
x = np.linspace(0,1,100) 
y = x**2 

import matplotlib.pyplot as plt 
plt.plot(x, y, label = '{\\footnotesize \$y = x^2\$}') 
plt.legend(loc = 'best') 
plt.show() 

Problem polega na tym, jak widać, że wyrównanie i rozmiar pudełka wokół legendy jest nieprawidłowy. Dzieje się tak, ponieważ rozmiar tekstu etykiety zmienia się, gdy obraz jest przesyłany przez Inkscape + pdflatex (ponieważ \footnotesize itd. Znika i zmienia się rozmiar czcionki).

I zorientowali się, że mogę wybrać położenie etykiety przez którąkolwiek

plt.label(loc = 'upper right') 

lub jeśli chcesz mieć większą kontrolę mogę używać

plt.label(bbox_to_anchor = [0.5, 0.2]) 

ale nie znalazłem żadnego sposobu zmniejszania pola na etykiecie. czy to możliwe?

Alternatywą do podejmowania pole mniejsze jest usunięcie obrys skrzynki używając coś jak

legend = plt.legend() 
legend.get_frame().set_edgecolor('1.0') 

a następnie przesuwając etykietę gdzie chcę. W takim razie chciałbym być w stanie określić położenie etykiety przez pierwszy najmu python/matplotlib go umieścić za pomocą

plt.label(loc = 'upper right') 

a następnie na przykład przesuwając go nieco w prawo. czy to możliwe? Próbowałem używać get_bbox_to_anchor() i set_bbox_to_anchor(), ale nie mogę go uruchomić.

Answer 1

Możesz przenieść legendę po jej automatycznym umieszczeniu przez narysowanie, a następnie uzyskanie pozycji boksa. Oto przykład:

import matplotlib.pyplot as plt 
import numpy as np 

# Plot data 
x = np.linspace(0,1,100) 
y = x**2 
fig = plt.figure() 
ax = fig.add_subplot(221) #small subplot to show how the legend has moved. 

# Create legend 
plt.plot(x, y, label = '{\\footnotesize \$y = x^2\$}') 
leg = plt.legend(loc = 'upper right') 

plt.draw() # Draw the figure so you can find the positon of the legend. 

# Get the bounding box of the original legend 
bb = leg.get_bbox_to_anchor().inverse_transformed(ax.transAxes) 

# Change to location of the legend. 
xOffset = 1.5 
bb.x0 += xOffset 
bb.x1 += xOffset 
leg.set_bbox_to_anchor(bb, transform = ax.transAxes) 


# Update the plot 
plt.show() 

Answer 2

Można użyć parametrów bbox_to_anchor i bbox_transform pomóc ustawienie kotwicę do legendy:

ax = plt.gca() 
plt.legend(bbox_to_anchor=(1.1, 1.1), bbox_transform=ax.transAxes) 

Zauważ, że (1.1, 1.1) są w osiach współrzędnych w ten przykład. Jeśli chcesz użyć współrzędnych danych, musisz użyć wartości bbox_transform=ax.transData.