I could comprehend the first piece of code in about 20 seconds and the final result in a couple minutes. It's pulled apart nicely, but it's not as simple any more.
The end result might look better in something like Lisp or Ruby, but I don't think it's a great tradeoff in Python.