Cognite interview question

Python Coding to find some numbers.