Entfernen von Leerzeichen um ein gespeichertes Bild in matplotlib

Ich muss ein Bild aufnehmen und es nach einiger Zeit speichern. Die Figur sieht gut aus, wenn ich sie anzeige, aber nach dem Speichern der Figur habe ich einen Leerraum um das gespeicherte Bild. Ich habe die 'tight'Option für savefigMethode ausprobiert , hat auch nicht funktioniert. Der Code:

  import matplotlib.image as mpimg
  import matplotlib.pyplot as plt

  fig = plt.figure(1)
  img = mpimg.imread(path)
  plt.imshow(img)
  ax=fig.add_subplot(1,1,1)

  extent = ax.get_window_extent().transformed(fig.dpi_scale_trans.inverted())
  plt.savefig('1.png', bbox_inches=extent)

  plt.axis('off') 
  plt.show()

Ich versuche, ein grundlegendes Diagramm mithilfe von NetworkX für eine Figur zu zeichnen und zu speichern. Ich habe festgestellt, dass es ohne Grafik funktioniert, aber wenn ich eine Grafik hinzufüge, bekomme ich einen Leerraum um das gespeicherte Bild.

import matplotlib.image as mpimg
import matplotlib.pyplot as plt
import networkx as nx

G = nx.Graph()
G.add_node(1)
G.add_node(2)
G.add_node(3)
G.add_edge(1,3)
G.add_edge(1,2)
pos = {1:[100,120], 2:[200,300], 3:[50,75]}

fig = plt.figure(1)
img = mpimg.imread("C:\\images\\1.jpg")
plt.imshow(img)
ax=fig.add_subplot(1,1,1)

nx.draw(G, pos=pos)

extent = ax.get_window_extent().transformed(fig.dpi_scale_trans.inverted())
plt.savefig('1.png', bbox_inches = extent)

plt.axis('off') 
plt.show()

Ich kann nicht behaupten, dass ich genau weiß, warum oder wie meine „Lösung“ funktioniert, aber das musste ich tun, als ich den Umriss einiger Tragflächenabschnitte - ohne weiße Ränder - in eine PDF-Datei zeichnen wollte. (Beachten Sie, dass ich matplotlib in einem IPython-Notizbuch mit dem Flag -pylab verwendet habe.)

plt.gca().set_axis_off()
plt.subplots_adjust(top = 1, bottom = 0, right = 1, left = 0, 
            hspace = 0, wspace = 0)
plt.margins(0,0)
plt.gca().xaxis.set_major_locator(plt.NullLocator())
plt.gca().yaxis.set_major_locator(plt.NullLocator())
plt.savefig("filename.pdf", bbox_inches = 'tight',
    pad_inches = 0)

Ich habe versucht, verschiedene Teile davon zu deaktivieren, aber dies führte immer irgendwo zu einem weißen Rand. Möglicherweise müssen Sie dies sogar ändern, um zu verhindern, dass fette Linien in der Nähe der Grenzen der Figur durch fehlende Ränder rasiert werden.

Sie können den weißen Raum Polsterung entfernen , indem Sie bbox_inches="tight"in savefig:

plt.savefig("test.png",bbox_inches='tight')

Sie müssen das Argument bbox_inchesals Zeichenfolge verwenden. Vielleicht hat es deshalb bei Ihnen früher nicht funktioniert.

Mögliche Duplikate:

Matplotlib-Diagramme: Entfernen von Achsen, Legenden und Leerzeichen

Wie stelle ich die Ränder für eine Matplotlib-Figur ein?

Reduzieren Sie den linken und rechten Rand im Matplotlib-Plot

Ich fand etwas von Arvind Pereira ( http://robotics.usc.edu/~ampereir/wordpress/?p=626 ) und schien für mich zu arbeiten:

plt.savefig(filename, transparent = True, bbox_inches = 'tight', pad_inches = 0)

Nachdem ich die obigen Antworten ohne Erfolg ausprobiert hatte (und eine Menge anderer Stapelpfosten), funktionierte das, was schließlich für mich funktionierte, einfach

plt.gca().set_axis_off()
plt.subplots_adjust(top = 1, bottom = 0, right = 1, left = 0, 
            hspace = 0, wspace = 0)
plt.margins(0,0)
plt.savefig("myfig.pdf")

Wichtig ist, dass dies nicht die Argumente bbox oder padding enthält.

Die folgende Funktion enthält die obige Antwort von johannes. Ich habe es mit plt.figureund plt.subplots()mit mehreren Achsen getestet und es funktioniert gut.

def save(filepath, fig=None):
    '''Save the current image with no whitespace
    Example filepath: "myfig.png" or r"C:\myfig.pdf" 
    '''
    import matplotlib.pyplot as plt
    if not fig:
        fig = plt.gcf()

    plt.subplots_adjust(0,0,1,1,0,0)
    for ax in fig.axes:
        ax.axis('off')
        ax.margins(0,0)
        ax.xaxis.set_major_locator(plt.NullLocator())
        ax.yaxis.set_major_locator(plt.NullLocator())
    fig.savefig(filepath, pad_inches = 0, bbox_inches='tight')

Ich fand, dass die folgenden Codes perfekt für den Job funktionieren.

fig = plt.figure(figsize=[6,6])
ax = fig.add_subplot(111)
ax.imshow(data)
ax.axes.get_xaxis().set_visible(False)
ax.axes.get_yaxis().set_visible(False)
ax.set_frame_on(False)
plt.savefig('data.png', dpi=400, bbox_inches='tight',pad_inches=0)

Ich folgte dieser Sequenz und es funktionierte wie ein Zauber.

plt.axis("off")
fig=plt.imshow(image array,interpolation='nearest')
fig.axes.get_xaxis().set_visible(False)
fig.axes.get_yaxis().set_visible(False)
plt.savefig('destination_path.pdf',
    bbox_inches='tight', pad_inches=0, format='pdf', dpi=1200)

Für alle, die lieber in Pixel als in Zoll arbeiten möchten, funktioniert dies.

Plus das Übliche, das Sie auch brauchen werden

from matplotlib.transforms import Bbox

Dann können Sie Folgendes verwenden:

my_dpi = 100 # Good default - doesn't really matter

# Size of output in pixels
h = 224
w = 224

fig, ax = plt.subplots(1, figsize=(w/my_dpi, h/my_dpi), dpi=my_dpi)

ax.set_position([0, 0, 1, 1]) # Critical!

# Do some stuff
ax.imshow(img)
ax.imshow(heatmap) # 4-channel RGBA
ax.plot([50, 100, 150], [50, 100, 150], color="red")

ax.axis("off")

fig.savefig("saved_img.png",
            bbox_inches=Bbox([[0, 0], [w/my_dpi, h/my_dpi]]),
            dpi=my_dpi)

Ein viel einfacherer Ansatz, den ich gefunden habe, ist plt.imsave:

    import matplotlib.pyplot as plt
    arr = plt.imread(path)
    plt.imsave('test.png', arr)

Wenn Sie anzeigen möchten, was gespeichert werden soll, empfehle ich die Verwendung der plt.tight_layoutTransformation, die eigentlich vorzuziehen ist, da bei der Verwendung kein unnötiges Zuschneiden erfolgtplt.savefig

import matplotlib as plt    
plt.plot([1,2,3], [1,2,3])
plt.tight_layout(pad=0)
plt.savefig('plot.png')

Dies funktioniert für mich beim Speichern eines numpy-Arrays, das mit imshow to file gezeichnet wurde

import matplotlib.pyplot as plt

fig = plt.figure(figsize=(10,10))
plt.imshow(img) # your image here
plt.axis("off")
plt.subplots_adjust(top = 1, bottom = 0, right = 1, left = 0, 
        hspace = 0, wspace = 0)
plt.savefig("example2.png", box_inches='tight', dpi=100)
plt.show()