Como alterar o tamanho de figuras criadas com Matplotlib

Ao trabalhar com Matplotlib, uma biblioteca Python popular para visualização de dados, você pode se ver na necessidade de ajustar o tamanho de suas figuras para que se encaixem melhor na apresentação dos seus dados ou no layout da sua publicação. Neste post, vamos explorar vários métodos para alterar o tamanho de figuras criadas com Matplotlib.

Definindo o Tamanho da Figura na Criação

A maneira mais direta de definir o tamanho de uma figura no Matplotlib é no momento da sua criação. Você pode especificar o tamanho da figura em polegadas usando o parâmetro figsize e definir a resolução com o parâmetro dpi (pontos por polegada). Veja como você pode fazer isso:

from matplotlib.pyplot import figure

# Criar uma figura com um tamanho e resolução específicos
figure(figsize=(8, 6), dpi=80)

Por exemplo, se você quiser criar uma figura que tenha uma polegada por uma polegada, você usaria figure(figsize=(1,1)). Se você não especificar o dpi, o padrão será de 80 pixels por polegada.

Ajustando o Tamanho de uma Figura Existente

Se você já criou uma figura e quer mudar seu tamanho, você pode usar o método set_size_inches. Este método permite ajustar o tamanho da figura em polegadas:

import matplotlib.pyplot as plt

# Obter a figura atual
fig = plt.gcf()
# Definir o tamanho da figura
fig.set_size_inches(18.5, 10.5)
# Salvar a figura com um novo tamanho
fig.savefig('test2png.png', dpi=100)

Para garantir que a mudança de tamanho afete uma janela de GUI existente, você pode adicionar o parâmetro forward=True:

fig.set_size_inches(18.5, 10.5, forward=True)

Você também pode mudar a resolução da figura usando set_dpi:

fig.set_dpi(100)

Alterando o Tamanho da Figura para Plotagem Inline

Ao plotar inline, como em um IPython Notebook, você pode mudar o tamanho da figura sem usar o ambiente figure ao definir o parâmetro figure.figsize em plt.rcParams:

import matplotlib.pyplot as plt

# Definir o tamanho padrão da figura
plt.rcParams["figure.figsize"] = (20, 3)

Lembre-se de evitar colocar essa instrução na mesma célula que seus comandos de importação. Para redefinir o tamanho da figura para o padrão em gráficos subsequentes, use:

plt.rcParams["figure.figsize"] = plt.rcParamsDefault["figure.figsize"]

Conversão para Centímetros

Se você prefere trabalhar em centímetros, pode converter o tamanho para polegadas dividindo os valores em centímetros por 2,54:

# Definir o tamanho da figura em centímetros
largura_cm = 10
altura_cm = 20
fig.set_size_inches(largura_cm / 2.54, altura_cm / 2.54)

Nota sobre Depreciação

É importante notar que o uso do módulo pylab está depreciado. Em vez disso, você deve usar o módulo matplotlib.pyplot. Se você encontrar código antigo que usa pylab, veja como ele pode parecer:

from pylab import rcParams
rcParams['figure.figsize'] = 5, 10

Isso definiria a largura da figura para 5 polegadas e sua altura para 10 polegadas.

Resumo

Neste post, cobrimos como alterar o tamanho de figuras no Matplotlib, seja criando uma nova figura ou ajustando uma já existente. Também discutimos como definir o tamanho da figura para plotagem inline e como converter medidas de centímetros para polegadas. Lembre-se de usar o módulo matplotlib.pyplot e evitar o módulo depreciado pylab. Com essas dicas, você pode garantir que suas visualizações tenham o tamanho perfeito para suas necessidades.