Pergunta de entrevista da empresa Data Axle

converting java code to python