Enregistrer le tracé dans un fichier image au lieu de l'afficher

Question

Cela permet d'afficher la figure dans une interface graphique :

import matplotlib.pyplot as plt
plt.plot([1, 2, 3], [1, 4, 9])
plt.show()

Mais comment puis-je enregistrer la figure dans un fichier (par exemple foo.png) ?

Answer 1

Vous pouvez procéder de la manière suivante :

def plotAFig():
  plt.figure()
  plt.plot(x,y,'b-')
  plt.savefig("figurename.png")
  plt.close()

Answer 2

Rien ne fonctionnait pour moi. Le problème est que l'image sauvegardée était très petite et je n'arrivais pas à trouver comment l'agrandir.

Cela semble l'agrandir, mais il ne s'agit toujours pas d'un plein écran.

https://matplotlib.org/stable/api/figure_api.html#matplotlib.figure.Figure.set_size_inches

fig.set_size_inches((w, h))

J'espère que cela aidera quelqu'un.

Answer 3

Vous pouvez enregistrer votre image avec n'importe quelle extension (png, jpg, etc.) et avec la résolution que vous souhaitez. Voici une fonction pour sauvegarder votre image.

import os

def save_fig(fig_id, tight_layout=True, fig_extension="png", resolution=300):
    path = os.path.join(IMAGES_PATH, fig_id + "." + fig_extension)
    print("Saving figure", fig_id)
    if tight_layout:
        plt.tight_layout()
    plt.savefig(path, format=fig_extension, dpi=resolution)

fig_id' est le nom sous lequel vous souhaitez enregistrer votre figure. J'espère que cela vous aidera :)

Answer 4

Utilise 'agg' car il n'y a pas d'interface utilisateur sur le serveur. Débogage sur ubuntu 21.10 avec gui et VSC. Dans le débogage, j'essaie à la fois d'afficher un tracé et de l'enregistrer dans un fichier pour l'interface web.

J'ai découvert qu'il est nécessaire de sauvegarder avant d'afficher, sinon la parcelle sauvegardée est vierge. Je suppose que l'affichage efface le tracé pour une raison quelconque. Voici ce qu'il faut faire :

plt.savefig(imagePath) 
plt.show()
plt.close(fig)

Au lieu de cela :

plt.show()
plt.savefig(imagePath) 
plt.close(fig)