Python code test - CARTO

process.py

#!/usr/local/bin/python3
line_count = 1
tip_amount_sum = 0

def process_line(lines):
  global line_count
  global tip_amount_sum
  line_count += 1
  tip_amount_sum += float(line.split(",")[15])

with open('yellow_tripdata_2016-01.csv', 'r') as trip_data:
  for line in trip_data:
    if 'tip_amount' in line:
      continue
    else:
      process_line(line)

print('lines: {}'.format(str(line_count)))
print('tip_amount average: ${}'.format(str(round(float(tip_amount_sum/(line_count-1)),2))))

python_code_test_carto.md

What follows is a technical test for this job offer at CARTO: https://boards.greenhouse.io/cartodb/jobs/705852#.WSvORxOGPUI

Build the following and make it run as fast as you possibly can using Python 3 (vanilla). The faster it runs, the more you will impress us!

Your code should:

• Download this ~2GB file: https://s3.amazonaws.com/carto-1000x/data/yellow_tripdata_2016-01.csv
• Count the lines in the file
• Calculate the average value of the tip_amount field.

All of that in the most efficient way you can come up with.

That's it. Make it fly!