Using Lambda Function In List Comprehensions Python Interview