Using Lambda Function In List Comprehensions In Python