import json

cfpb_16 = [x for x in json.load(open("cfpb16_1000.json"))]

# Task 1
def count_complaints(complaints, company_name):

    compcount = 0
    for complaint in complaints:
        if complaint["Company"] == company_name:
            compcount += 1
    return compcount

# Task 2
def find_companies(complaints):

    complaintset = set()
    for complaint in complaints:
        complaintset.add(complaint["Company"])
    return complaintset


# Task 3
def count_by_state(complaints):

    statedict = {}

    for complaint in complaints:
        statename = complaint["State"]
        statedict[statename] = statedict.get(statename, 0) + 1
    return statedict

# Task 4
def count_by_company_by_state(complaints):

    bcbs = {}
    for company_name in find_companies(complaints):

        cstate = {}
        for complaint in complaints:
            if complaint["Company"] == company_name:
                cstate[complaint["State"]] = cstate.get(complaint["State"], 0) + 1

        bcbs[company_name] = cstate
    return bcbs


# Task 5
def complaints_by_company(complaints):
    # Your code goes here
    # replace {} with a suitable return value
    return {}