Python Walrus Inside List Comprehension