Shift all indices in NumPy array

Question

I have a numpy array like this:

x=np.array([0,1,2,3,4])

and want to create an array where the value in index 0 is in index 1, index 1 is in index 2, etc.

The output I want is:

y=np.array([0,0,1,2,3]).

I'm guessing there's an easy way to do this without iterating through the full array. How can I do this in a numPythonic way?

Sven Marnach · Accepted Answer · 2011-05-02 15:41:37Z

17

You can use

y = numpy.roll(x, 1)
y[0] = 0

or

y = numpy.r_[0, x[:-1]]

answered May 2, 2011 at 15:41

Sven Marnach

608k123 gold badges966 silver badges865 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

1'' · Accepted Answer · 2016-05-21 09:59:25Z

3

If you wanted to do this in-place, you could also do:

x[1:] = x[:-1]
x[0] = 0

answered May 21, 2016 at 9:59

1''

27.2k32 gold badges149 silver badges204 bronze badges