EY interview question

Write a simple python code.