Como salvar uma figura com diferentes tamanhos

Antes de salvar a visualização, talvez seja conveniente definir o tamanho que a figura terá na página. Para fazer isso, você pode usar o método set_size_inches do objeto Figure. Esse método usa uma sequência de dois valores. O primeiro define a largura, e o segundo define a altura da figura.

Aqui você já recebeu novamente um objeto Figure chamado fig (se quiser ver o conteúdo dele, você pode executar plt.show). Use os métodos Figure set_size_inches e savefig para alterar o tamanho e salvar duas versões diferentes dessa figura.