NumPy - Insert an array of zeros after specified indices

Question

My code is:

x=np.linspace(1,5,5)

a=np.insert(x,np.arange(1,5,1),np.zeros(3))

The output I want is:

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

The error I get is:

ValueError: shape mismatch: value array of shape (3,) could not be broadcast to indexing result of shape (4,)

When I do:

x=np.linspace(1,5,5)

a=np.insert(x,np.arange(1,5,1),0)

The out is:

array([1., 0., 2., 0., 3., 0., 4., 0., 5.])

Why it doesn't work when I try to insert an array?

P.S. I I cannot use loops

jpp · Accepted Answer · 2018-11-08 14:25:40Z

3

You can use np.repeat to feed repeated indices. For a 1d array, thhe obj argument for np.insert reference individual indices.

x = np.linspace(1, 5, 5)

a = np.insert(x, np.repeat(np.arange(1, 5, 1), 3), 0)

array([ 1.,  0.,  0.,  0.,  2.,  0.,  0.,  0.,  3.,  0.,  0.,  0.,  4.,
        0.,  0.,  0.,  5.])

edited Nov 8, 2018 at 14:25

answered Nov 8, 2018 at 13:16

jpp

166k37 gold badges301 silver badges363 bronze badges

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

Comments

xnx · Accepted Answer · 2018-11-08 13:19:20Z

2

Another option:

np.hstack((x[:,None], np.zeros((5,3)))).flatten()[:-3]

gives:

array([ 1.,  0.,  0.,  0.,  2.,  0.,  0.,  0.,  3.,  0.,  0.,  0.,  4.,
    0.,  0.,  0.,  5.])

That is, pretend x is a column vector and stack a 5x3 block of zeros to the right of it and then flatten.

answered Nov 8, 2018 at 13:19

xnx

25.7k11 gold badges75 silver badges118 bronze badges

Collectives™ on Stack Overflow

NumPy - Insert an array of zeros after specified indices

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related