List comprehension in Swift with lazy evaluation / deferred execution