Bekommst die Legende als separates Bild in Matplotlib

Ich bin der Entwicklung einer Web-Anwendung und möchten Sie ein Bild und seine Legende, die in verschiedenen Orten auf der Seite. Das heißt, ich brauche zum speichern der Legende als separate png-Datei. Ist das möglich in der Matplotlib in einem mehr oder weniger einfachen Art und Weise?

InformationsquelleAutor der Frage Ruggiero Spearman | 2010-12-26

  1. 18

    Das funktionieren könnte:

    import pylab
fig = pylab.figure()
figlegend = pylab.figure(figsize=(3,2))
ax = fig.add_subplot(111)
lines = ax.plot(range(10), pylab.randn(10), range(10), pylab.randn(10))
figlegend.legend(lines, ('one', 'two'), 'center')
fig.show()
figlegend.show()
figlegend.savefig('legend.png')

    Bekommst die Legende als separates Bild in Matplotlib

    InformationsquelleAutor der Antwort Steve Tjoa

  2. 11

    verwenden pylab.figlegend(..) und get_legend_handles_labels(..):

    import pylab, numpy 
x = numpy.arange(10)

# create a figure for the data
figData = pylab.figure()
ax = pylab.gca()

for i in xrange(3):
    pylab.plot(x, x * (i+1), label='line %d' % i)

# create a second figure for the legend
figLegend = pylab.figure(figsize = (1.5,1.3))

# produce a legend for the objects in the other figure
pylab.figlegend(*ax.get_legend_handles_labels(), loc = 'upper left')

# save the two figures to files
figData.savefig("plot.png")
figLegend.savefig("legend.png")

    Es kann schwierig sein, wenn man die Größe der Legende der Abbildung rechts in einer automatisierten Weise.

    InformationsquelleAutor der Antwort Andre Holzner

  3. 6

    Dieser berechnet die Größe der Legende automatisch. Wenn mode == 1der code ist ähnlich wie Steve Tjoa Antwort, während mode == 2 ist ähnlich wie Andre Holzner Antwort.

    Den loc parameter muss eingestellt werden 'center' machen, damit es funktioniert (aber ich weiß nicht, warum dies notwendig ist).

    mode = 1
#mode = 2

import pylab
fig = pylab.figure()
if mode == 1:
    lines = fig.gca().plot(range(10), pylab.randn(10), range(10), pylab.randn(10))
    legend_fig = pylab.figure(figsize=(3,2))
    legend = legend_fig.legend(lines, ('one', 'two'), 'center')
if mode == 2:
    fig.gca().plot(range(10), pylab.randn(10), range(10), pylab.randn(10), label='asd')
    legend_fig = pylab.figure()
    legend = pylab.figlegend(*fig.gca().get_legend_handles_labels(), loc = 'center')
legend.get_frame().set_color('0.70')
legend_fig.canvas.draw()
legend_fig.savefig('legend_cropped.png',
    bbox_inches=legend.get_window_extent().transformed(legend_fig.dpi_scale_trans.inverted()))
legend_fig.savefig('legend_original.png')

    Original (unbeschnitten) Legende:

    Bekommst die Legende als separates Bild in Matplotlib

    Beschnitten Legende:

    Bekommst die Legende als separates Bild in Matplotlib

    InformationsquelleAutor der Antwort klaus se

  4. 4

    Können Sie begrenzen die gesicherte region der ein Bild auf den Begrenzungsrahmen der Legende mit der bbox_inches argument fig.savefig. Unten, um Versionen einer Funktion, die Sie einfach aufrufen kann mit der Legende, die Sie speichern möchten als argument. Sie können entweder verwenden Sie die Legende erstellt, in der ursprünglichen Gestalt (und entfernen Sie es danach legend.remove()) oder Sie erstellen eine neue Figur für die Legende und verwenden Sie einfach die Funktion wie es ist.

    Export Legende boundingbox

    Im Falle der vollständigen Legende soll gerettet werden, die bounding-box geliefert, um die bbox_inches argument wäre einfach die transformierten bounding box der Legende. Dies funktioniert gut, wenn die Legende hat keine Grenze um ihn herum.

    import matplotlib.pyplot as plt

colors = ["crimson", "purple", "gold"]
f = lambda m,c: plt.plot([],[],marker=m, color=c, ls="none")[0]
handles = [f("s", colors[i]) for i in range(3)]
labels = colors
legend = plt.legend(handles, labels, loc=3, framealpha=1, frameon=False)

def export_legend(legend, filename="legend.png"):
    fig  = legend.figure
    fig.canvas.draw()
    bbox  = legend.get_window_extent().transformed(fig.dpi_scale_trans.inverted())
    fig.savefig(filename, dpi="figure", bbox_inches=bbox)

export_legend(legend)
plt.show()

    Bekommst die Legende als separates Bild in Matplotlib

    Export erweitert Legende bounding-box

    Wenn es einen Rahmen um die Legende, die oben genannte Lösung kann suboptimal. In diesem Fall macht es Sinn zu erweitern, die bounding box durch einige Pixel zu zählen der Grenze zu seine volle.

    import numpy as np
import matplotlib.pyplot as plt

colors = ["crimson", "purple", "gold"]
f = lambda m,c: plt.plot([],[],marker=m, color=c, ls="none")[0]
handles = [f("s", colors[i]) for i in range(3)]
labels = colors
legend = plt.legend(handles, labels, loc=3, framealpha=1, frameon=True)

def export_legend(legend, filename="legend.png", expand=[-5,-5,5,5]):
    fig  = legend.figure
    fig.canvas.draw()
    bbox  = legend.get_window_extent()
    bbox = bbox.from_extents(*(bbox.extents + np.array(expand)))
    bbox = bbox.transformed(fig.dpi_scale_trans.inverted())
    fig.savefig(filename, dpi="figure", bbox_inches=bbox)

export_legend(legend)
plt.show()

    Bekommst die Legende als separates Bild in Matplotlib

    InformationsquelleAutor der Antwort ImportanceOfBeingErnest

  5. 1

    Ist es möglich, Achsen.get_legend_handles_labels zu Holen Sie sich die Legende, Griffe und Etiketten aus einer axes Objekt und verwenden Sie Sie, um Sie hinzuzufügen, um eine Achse in eine andere Figur.

    # create a figure with one subplot
fig = plt.figure()
ax = fig.add_subplot(111)
ax.plot([1,2,3,4,5], [1,2,3,4,5], 'r', label='test')
# save it *without* adding a legend
fig.savefig('image.png')

# then create a new image
# adjust the figure size as necessary
figsize = (3, 3)
fig_leg = plt.figure(figsize=figsize)
ax_leg = fig_leg.add_subplot(111)
# add the legend from the previous axes
ax_leg.legend(*ax.get_legend_handles_labels(), loc='center')
# hide the axes frame and the x/y labels
ax_leg.axis('off')
ax_leg.savefig('legend.png')

    Wenn Sie aus irgendeinem Grund ausblenden möchten nur die Achsen, die Sie verwenden können:

    ax.xaxis.set_visible(False)
ax.yaxis.set_visible(False)

    oder wenn Sie für einige seltsamer Grund, Sie zu verstecken, Achsen Rahmen, aber nicht die Achsen Etiketten, die Sie verwenden können:

    ax.set_frame_on(False)

    ps: diese Antwort wurde angepasst, aus meiner Antwort auf eine doppelte Frage

    InformationsquelleAutor der Antwort Francesco Montesano

Schreibe einen Kommentar