expand 1d numpy array

Question

I want to expand/repeat each element in a 1D array for different times

x = array([7,1,9])  #1d array
rep_size = array([3,2,2]) # repeat number for each element in x
result = arary([7,7,7,1,1,9,9]) #expected result

Is there a numpy function can do this if I don't want to use a for loop. Thanks.

mozway · Accepted Answer · 2022-05-18 14:02:00Z

1

result = np.repeat(x, rep_size)

output: array([7, 7, 7, 1, 1, 9, 9])

answered May 18, 2022 at 14:02

mozway

267k13 gold badges56 silver badges106 bronze badges

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

1 Answer 1