Como salvar a figura matplotlib em um arquivo (com exemplos)

Você pode usar a seguinte sintaxe básica para salvar uma figura Matplotlib em um arquivo:

 import matplotlib. pyplot as plt

#save figure in various formats
plt. savefig (' my_plot.png ')
plt. savefig ( ' my_plot.jpg ') 
plt. savefig ( ' my_plot.pdf ')

Os exemplos a seguir mostram como usar essa sintaxe na prática.

Exemplo 1: Salve a figura Matplotlib em um arquivo PNG

O código a seguir mostra como salvar uma figura Matplotlib em um arquivo PNG:

 import matplotlib. pyplot as plt

#define data
x = [1, 2, 3, 4, 5, 6]
y = [8, 13, 14, 11, 16, 22]

#create scatterplot with axis labels
plt. plot (x, y)
plt. xlabel (' XVariable ')
plt. ylabel (' Y Variable ')

#save figure to PNG file
plt. savefig (' my_plot.png ')

Se navegarmos até o local onde salvamos o arquivo, podemos visualizá-lo:

Exemplo 2: Salvar figura Matplotlib com layout compacto

Por padrão, o Matplotlib adiciona um preenchimento generoso na parte externa da figura.

Para remover esse preenchimento, podemos usar o argumento bbox_inches=’tight’ :

 #save figure to PNG file with no padding
plt. savefig (' my_plot.png ', bbox_inches=' tight ') 

Observe que há menos preenchimento fora do gráfico.

Exemplo 3: Salvar figura Matplotlib com tamanho personalizado

Você também pode usar o argumento dpi para aumentar o tamanho da figura Matplotlib ao salvá-la:

 #save figure to PNG file with increased size
plt. savefig (' my_plot.png ', dpi= 100 ) 

Você pode encontrar a documentação online completa da função Matplotlib savefig() aqui .

Recursos adicionais

Os tutoriais a seguir explicam como executar outras funções comuns no Matplotlib:

Como definir intervalos de eixos no Matplotlib
Como aumentar o tamanho do gráfico no Matplotlib
Como criar vários gráficos Matplotlib em uma única figura