11_01_grouped_count.py

# coding: utf-8

from decimal import Decimal

def get_id_and_value(info, lower):
    value = Decimal(info[5])
    if value > lower:
        return info[2], value
    return None, Decimal(0)

all_companies = set()
intervals = [(Decimal('1000000000'), set()),
             (Decimal('500000000'), set()),
             (Decimal('100000000'), set()),
             (Decimal('10000000'), set()),
             (Decimal('1000000'), set()),
             (Decimal('100000'), set()),
             (Decimal('10000'), set()),
             (Decimal('1000'), set())]

for lower, companies in intervals:
    data = open('data/data/ExecucaoFinanceira.csv', 'r')
    for line in data:
        company_id, contract_value = get_id_and_value(
            line.strip().split(';'), lower)
        if company_id and not company_id in all_companies:
            companies.add(company_id)
        all_companies.add(company_id)
    data.close()

for lower, companies in intervals:
    print("{} empresas receberam mais de {}".format(
        len(companies), lower))
print("{} empresas no total".format(len(all_companies)))