Use loop comprehension instead of map().

Generally, loop comprehensions look nicer in Python than `lambda`. And they are shorter, in case someone wants to code golf this, or something.