Pregunta de entrevista de Data Axle

converting java code to python