Tutorial Matplotlib

Configuração inicial

Instalação

pip install matplotlib

Código Básico

Para fins didáticos, utilizarei x e y criados com valores aleatórios. Sendo para cada x um y correspondente.

x = [1, 3, 6, 8, 11, 13, 17, 22]
y = [10, 20, 25, 15, 30, 23, 17, 27]

E então precisa 'plotar' o gráfico, ou seja, adicionar esse gráfico formado com esses valores ao escopo

plt.plot(x, y)

E então, mostra-se na tela a figura gerada:

plt.show()

Assim, teremos:

Adicionando Labels

Para facilitar a compreensão do gráfico, coloca-se título e o que cada eixo representa:

plt.title("Temperaturas de Bertolândia em Outubro")

plt.xlabel("Dias")
plt.ylabel("Temperatura (°C)")

Melhorando muito mais a visualização e entendimento das informações contidas nele:

Adicionando um novo dado para comparação

Lembra que comentei do plot()? Que você colocaria no gráfico os dados e depois mostraria na tela com o show()?

Agora colocaremos isso em prática!

Vamos alterar algumas coisas, vamos trocar alguns dados e nomes de algumas variáveis:

dias = [1, 3, 6, 8, 11, 13, 17, 22]
temp_max = [22, 20, 25, 15, 30, 23, 17, 27]
temp_min = [17, 15, 18, 10, 22, 20, 8, 18]

plt.title("Temperaturas Máximas e Mínimas de Bertolândia em Outubro")

E agora vamos adicionar tanto as temperaturas máximas quanto as mínimas com o plot() e mostrar o resultado:

plt.plot(dias, temp_max)
plt.plot(dias, temp_min)
plt.show()

Adicionando Grid

Para adicionar um grid (fundo quadriculado) para melhorar a visualização pode-se usar:

plt.grid(True)

Estilização das linhas

Linestyle

Coloca-se no .plot

plt.plot(dias, temp_max, linestyle='--')

Marker

plt.plot(dias, temp_max, linestyle='--', marker='o')

Cor

plt.plot(dias, temp_min, color='#9407F2')

Linewidth

plt.plot(dias, temp_min, color='#9407F2', linewidth=5)

Legend

Como usar

Existem diversas formas de colocar o legend, mas a melhor delas é deixar um label no plot()

plt.plot(dias, temp_max, linestyle='--', marker='o', label="Temperatura Máxima")
plt.plot(dias, temp_min, color='#9407F2', linewidth=5, label="Temperatura Mínima")

plt.legend()

Mudando a posição do legend

Mas também pode-se colocar a legenda em diferentes locais da tela usando o parâmetro loc

E você pode usar diversas localizações como:

best (O próprio programa decide o melhor local)

upper right

upper left

lower left

lower right

right

center left

center right

lower center

upper center

center

E usa-se assim:

plt.legend(loc="best")

Definir intervalo do gráfico

#       ([Xmin, Xmax, Ymin, Ymax])
plt.axis([6, 18, 5, 35])

E pode-se também modificar apenas um dos eixos com xlim() e ylim()

plt.xlim([6, 18])
plt.ylim([5, 35])

Quando se faz apenas do X, o próprio matplot decide os melhores valores para colocar nos limites de y, e a inversa também ocorre o mesmo.

Salvar o Gráfico

Ao invés de dar um show() para mostrar o gráfico, pode-se salvar a imagem com:

plt.savefig(<file name>)

E irá aparecer junto aos arquivos