Collapse neat

A slightly neater version of pull request https://github.com/sazlin/more-itertools/pull/1 that is about 8-12% slower to use the resulting generator.