Matplotlib intermediário

Gráficos com DataFrames do pandas

Transforme um DataFrame do pandas em gráfico com df.plot e personalize com Matplotlib.

Introdução

O pandas vem com Matplotlib embutido: df.plot desenha um gráfico direto de um DataFrame e usa os nomes das colunas na legenda. Ele devolve um Axes do Matplotlib, então tudo o que você aprendeu continua valendo para títulos, rótulos e estilo.

x="month" coloca os meses no eixo x e y=[...] escolhe as colunas; kind escolhe o gráfico ("line", "bar", "barh", "area"...). df.plot devolve ax, o Axes normal do Matplotlib, então ax.set_ylabel e ax.set_title funcionam como antes. rot=0 deixa os meses na horizontal e stacked=True empilharia as barras.

Visualizador interativo

Mude o kind, escolha as colunas e empilhe-as. O pandas desenha com Matplotlib e devolve um Axes que você pode continuar personalizando.

Exemplo de Matplotlib executável

Matplotlib
Executar código →
main.py
import matplotlib.pyplot as plt
import pandas as pd

df = pd.DataFrame({
    "month": ["Jan", "Feb", "Mar", "Apr", "May", "Jun"],
    "coffee": [120, 110, 130, 150, 170, 160],
    "tea": [80, 85, 90, 70, 60, 65],
})

ax = df.plot(x="month", y=["coffee", "tea"], kind="bar", rot=0, figsize=(7, 4))
ax.set_ylabel("Cups sold")
ax.set_title("Drinks sold per month")
plt.show()
Gerar um botão Run para este exemplo O gráfico que este código desenha
Gráficos com DataFrames do pandas