web-dev-qa-db-fra.com

Matplotlib savefig avec une légende en dehors du complot

En lisant l'article suivant, j'ai réussi à mettre une légende en dehors de l'intrigue.

code:

import matplotlib.pyplot as pyplot

x = [0, 1, 2, 3, 4]
y = [xx*xx for xx in x]

fig = pyplot.figure()
ax  = fig.add_subplot(111)

box = ax.get_position()
ax.set_position([box.x0, box.y0, box.width*0.8, box.height])

ax.plot(x, y)
leg = ax.legend(['abc'], loc = 'center left', bbox_to_anchor = (1.0, 0.5))
#pyplot.show()

fig.savefig('aaa.png', bbox_inches='tight')

pyplot.show() affiche le tracé correct avec une légende à l'extérieur. Mais lorsque je l'enregistre dans un fichier avec fig.savefig(), la légende est tronquée.

Certaines recherches sur Google me proposent des solutions telles que l'ajout de bbox_extra_artists=[leg.legendPatch] ou bbox_extra_artists=[leg] à savefig(), mais aucune n'a fonctionné.

Quelle est la bonne façon de le faire? La version de Matplotlib est 0.99.3.

Merci.

pythonmatplotliblegend
31
niboshi

Le problème est que lorsque vous tracez de manière dynamique, matplotlib détermine automatiquement les bordures pour s’adapter à tous vos objets. Lorsque vous enregistrez un fichier, les choses ne se font pas automatiquement. Vous devez donc spécifier La taille de votre figure, puis le cadre de sélection de votre objet axes. Voici comment corriger votre code:

import matplotlib.pyplot as pyplot

x = [0, 1, 2, 3, 4]
y = [xx*xx for xx in x]

fig = pyplot.figure(figsize=(3,3))
ax  = fig.add_subplot(111)

#box = ax.get_position()
#ax.set_position([0.3, 0.4, box.width*0.3, box.height])
# you can set the position manually, with setting left,buttom, witdh, hight of the axis
# object
ax.set_position([0.1,0.1,0.5,0.8])
ax.plot(x, y)
leg = ax.legend(['abc'], loc = 'center left', bbox_to_anchor = (1.0, 0.5))

fig.savefig('aaa.png')
24
Oz123

Bien que cette méthode fonctionne avec la légende, elle semble ne pas bien fonctionner avec figlegend quand il y a plusieurs intrigues secondaires et que nous voulons une seule légende globale. figlegend est toujours recadré quand savefig. Je viens de coller ma solution temporaire ci-dessous au cas où quelqu'un ferait face à un tel cas.

import matplotlib.pyplot as plt

para = {
    ## this parameter will indicate the position of
    ## subplot within figure, but will not be shown
    ## if using bbox_inches='tight' when saving
    'figure.subplot.top': 0.5
}
#plt.rcParams.update(para)

fig = plt.figure()

ax=fig.add_subplot(221)
## only needed when what to manually control
## subplot ration
#ax.set_position([0.1,0.6,0.5, 0.4])
ax.plot([1,1,1])


ax=fig.add_subplot(222)
#ax.set_position([0.7,0.6,0.5, 0.4])
ax.plot([2,2,2])

ax=fig.add_subplot(223)
#ax.set_position([0.1,0.1,0.5, 0.4])
ax.plot([3,3,3])


ax=fig.add_subplot(224)
#ax.set_position([0.7,0.1,0.5, 0.4])
p1, = ax.plot([4,4,4])
p2, = ax.plot([2,3,2])

## figlegend does not work fine with tight bbox
## the legend always get cropped by this option
## even add bbox extra will not help
## had to use legend, and manually adjust it to
## arbitary position such as (0.3, 2.5)

## http://matplotlib.org/users/tight_layout_guide.html
## according to this link, tight layout is only
## an experimental feature, might not support figlegend

#lgd = plt.figlend(
lgd = plt.legend(
    [p1,p2],
    ['a', 'b'],
    ## by default, legend anchor to axis, but can
    ## also be anchored to arbitary position
    ## positions within [1,1] would be within the figure
    ## all numbers are ratio by default

    bbox_to_anchor=(-0.1, 2.5),

    ## loc indicates the position within the figure
    ## it is defined consistent to the same Matlab function 
    loc='center',

    ncol=2
    #mode="expand",
    #borderaxespad=0.
    )



#plt.show()

plt.savefig('temp.png', bbox_inches='tight')#, bbox_extra_artist=[lgd])
11
Ning

Si tout échoue, j'utilise les fonctionnalités de la boîte englobante d'Inkscape pour traiter ce que j'appellerais des bogues persistants dans la sortie de matplotlib. Si vous utilisez GNU/Linux, enregistrez tout ce que Matplotlib vous donne sous forme de fichier PDF, puis envoyez-le à l'adresse suivante:

def tightBoundingBoxInkscape(pdffile,use_xvfb=True):
    """Makes POSIX-specific OS calls. Preferably, have xvfb installed, to avoid any GUI popping up in the background. If it fails anyway, could always resort to use_xvfb=False, which will allow some GUIs to show as they carry out the task 
      pdffile: the path for a PDF file, without its extension
    """
    usexvfb='xvfb-run '*use_xvfb
    import os
    assert not pdffile.endswith('.pdf')
    os.system("""
       inkscape -f %(FN)s.pdf -l %(FN)s_tmp.svg
       inkscape -f %(FN)s_tmp.svg --verb=FitCanvasToDrawing \
                                   --verb=FileSave \
                                   --verb=FileQuit
      inkscape -f %(FN)s_tmp.svg -A %(FN)s-tightbb.pdf
"""%{'FN':pdffile}
0
CPBL