Case Study - US Births Analysis Function

.py

# write function for this analysis
def calculate_stat(data, k, v):
  """
  input: data, idx key, idx value
  output: dictionary with key: value pair
  """
  result = {}
  for row in data:
    key = row.split(",")[k] 
    value = int(row.split(",")[v])
    if key in result:
      result[key] += value
    else:
      result[key] = value
  return result