Contar, somar e agrupar dados

Introdução a Bancos de Dados em Python

Jason Myers

Co-Author of Essential SQLAlchemy and Software Engineer

Funções SQL

Ex.: COUNT, SUM
from sqlalchemy import func
Mais eficiente que processar em Python
Agregar dados

Exemplo de soma

from sqlalchemy import func

stmt = select([func.sum(census.columns.pop2008)])

results = connection.execute(stmt).scalar()

print(results)

302876613

Group by

Permite agrupar linhas por valores em comum

Group by

stmt = select([census.columns.sex, 
  func.sum(census.columns.pop2008)])

stmt = stmt.group_by(census.columns.sex)

results = connection.execute(stmt).fetchall()

print(results)

[('F', 153959198), ('M', 148917415)]

Group by

Suporta múltiplas colunas para agrupar, similar ao padrão de order_by()
Exige que todas as colunas selecionadas sejam agrupadas ou agregadas por uma função

Group by múltiplo

stmt = select([census.columns.sex,
        census.columns.age,
        func.sum(census.columns.pop2008)
    ])

stmt = stmt.group_by(census.columns.sex,
        census.columns.age)

results = connection.execute(stmt).fetchall()
print(results)

[('F', 0, 2105442), ('F', 1, 2087705), ('F', 2, 2037280),
('F', 3, 2012742), ('F', 4, 2014825), ('F', 5, 1991082),
('F', 6, 1977923), ('F', 7, 2005470), ('F', 8, 1925725), ...

Tratando ResultSets de funções

O SQLAlchemy gera automaticamente "nomes de coluna" para funções no ResultSet
Esses nomes costumam ser func_#, como count_1
Substitua-os com o método label()

Usando label()

print(results[0].keys())

['sex', u'sum_1']

stmt = select([census.columns.sex,
        func.sum(census.columns.pop2008).label('pop2008_sum')
    ])

stmt = stmt.group_by(census.columns.sex)

results = connection.execute(stmt).fetchall()
print(results[0].keys())

['sex', 'pop2008_sum']

Vamos praticar!

Introdução a Bancos de Dados em Python