python - Repeat different elements of an array different amounts of times

Question

Say I have an array with longitudes, lonPorts

lonPort =np.loadtxt('LongPorts.txt',delimiter=',')

for example:

lonPort=[0,1,2,3,...]

And I want to repeat each element a different amount of times. How do I do this? This is what I tried:

Repeat =[5, 3, 2, 3,...]

lonPort1=[]

for i in range (0,len(lenDates)):
   lonPort1[sum(Repeat[0:i])]=np.tile(lonPort[i],Repeat[i])

So the result would be:

lonPort1=[0,0,0,0,0,1,1,1,2,2,3,3,3,...]

The error I get is:

list assignment index out of range

How do I get rid of the error and make my array? Thank you!

Answer 1

You can use np.repeat():

np.repeat(a, [5,3,2,3])

Example:

In [3]: a = np.array([0,1,2,3])

In [4]: np.repeat(a, [5,3,2,3])
Out[4]: array([0, 0, 0, 0, 0, 1, 1, 1, 2, 2, 3, 3, 3])