Pergunta de entrevista da empresa DXC Technology

write a function in python