Preciso de uma janela rotativa (também conhecida como janela deslizante) iterável em uma sequência / iterador / gerador. A iteração padrão do Python pode ser considerada um caso especial, em que o comprimento da janela é 1. Atualmente, estou usando o código a seguir. Alguém tem um método mais pitonico, menos detalhado ou mais eficiente para fazer isso?

def rolling_window(seq, window_size):
    it = iter(seq)
    win = [it.next() for cnt in xrange(window_size)] # First window
    yield win
    for e in it: # Subsequent windows
        win[:-1] = win[1:]
        win[-1] = e
        yield win

if __name__=="__main__":
    for w in rolling_window(xrange(6), 3):
        print w

"""Example output:

   [0, 1, 2]
   [1, 2, 3]
   [2, 3, 4]
   [3, 4, 5]
"""