Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- 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 {}
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement