See also more-itertools for lots of other things in this vein.