average-attendance.py

import sys
import matplotlib.pyplot as plt
import csv

dados = 'data/average-attendance-2003-2019.txt'
ano_legenda = '2003-2019'
ano = None
if len(sys.argv) > 1:
    ano = sys.argv[1]
    ano_legenda = ano

clubes = {}
with open(dados, newline='', encoding='iso-8859-1') as arquivo:
    leitor = csv.reader(arquivo, delimiter=';')
    next(leitor) # pula a linha de cabeçalho
    for linha in leitor:
        if not linha:
            break # Se achar uma linha em branco, interrompe o processamento
        nome_clube = linha[1]
        media_publico = int(linha[2])
        if ano != None and linha[0] != ano:
            # Se tiver filtro por ano, ignora dados de outros anos
            continue
        if not nome_clube in clubes:
            clubes[nome_clube] = 0
        clubes[nome_clube] = clubes[nome_clube]+media_publico

# Ordena os clubes por público, descrescente
clubes = dict(sorted(clubes.items(), key=lambda item: item[1], reverse=True))

# Conta quantos clubes temos
ticks = list(range(0, len(clubes)))

# Prepara o nome do arquivo e título do gráfico
titulo = 'Média de público nos estádios em ' + ano_legenda
arquivo = 'average-attendance-'+ano_legenda+'.png'
print('Gerando o gráfico: ' + titulo + '...')

# Agora vamos gerar o gráfico
plt.title(titulo)
plt.bar(ticks, clubes.values())
plt.xticks(ticks, clubes.keys(), rotation='vertical')
plt.savefig(arquivo, dpi=150, bbox_inches='tight')

print('Gráfico gerado em ' + arquivo)