specific way of numpy array expansion

Question

I have the following numpy array (as an example):

[0,1,5,4,3]

Is there any way to "repeat" this array, but in a specific way: I need a final array of shape 25 but arranged in a way that firs I would have 5 zeros, then- 5 ones, then- 5 fives, etc. Example of desired output:

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

if I do np.append(arr, arr)- it will give me two initial sequenced arrays, and this is not my goal BTW, my real array is not sorted and should remain unsorted in the end.

Thanks in advance!

TrebledJ · Accepted Answer · 2019-04-06 04:53:17Z

1

Use np.repeat.

>>> np.repeat([0,1,5,4,3], 5)
array([0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 5, 5, 5, 5, 5, 4, 4, 4, 4, 4, 3, 3,
       3, 3, 3])

answered Apr 6, 2019 at 4:53

TrebledJ

9,0237 gold badges28 silver badges50 bronze badges

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

Collectives™ on Stack Overflow

specific way of numpy array expansion

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related