4-add_chart.py

from openpyxl import load_workbook
from openpyxl.chart import BarChart, Reference

# Ler pasta de trabalho e planilha
wb = load_workbook('data/pivot_table.xlsx')
sheet = wb['Relatorio']

# Referencias das linhas e colunas
min_column = wb.active.min_column
max_column = wb.active.max_column
min_row = wb.active.min_row
max_row = wb.active.max_row

# Adicionar dados e categorias no grafico
barchart = BarChart()

data = Reference(
    sheet,
    min_col = min_column + 1, # Sem o Ano
    max_col = max_column,
    min_row = min_row,
    max_row = max_row
)

categories = Reference(
    sheet,
    min_col = min_column,
    max_col = min_column,
    min_row = min_row + 1, # Sem o cabecalho
    max_row = max_row
)

barchart.add_data(data, titles_from_data = True)
barchart.set_categories(categories)

# Criar o grafico
sheet.add_chart(barchart, 'B10')
barchart.title = 'Vendas por Fabricantes'
barchart.style = 2

# Salvar o workbook
wb.save('data/barchart.xlsx')