python - Circular shift of vector (equivalent to numpy.roll)

Question

I have a vector:

a <- c(1,2,3,4,5)

And I'd like to do something like:

b <- roll(a, 2) # 4,5,1,2,3

Is there a function like that in R? I've been googling around, but "R Roll" mostly gives me pages about Spanish pronunciation.

Answer 1

How about using head and tail...

roll <- function( x , n ){
  if( n == 0 )
    return( x )
  c( tail(x,n) , head(x,-n) )
}

roll(1:5,2)
#[1] 4 5 1 2 3

#  For the situation where you supply 0 [ this would be kinda silly! :) ]
roll(1:5,0)
#[1] 1 2 3 4 5

One cool thing about using head and tail... you get a reverse roll with negative n, e.g.

roll(1:5,-2)
[1] 3 4 5 1 2