Create new numpy array by duplicating each item in an array

Question

I would like to create a new numpy array by repeating each item in another array by a given number of times (n). I am currently doing this with a for loop and .extend() with lists but this is not really efficient, especially for very large arrays.

Is there a more efficient way to do this?

def expandArray(array, n):
    new_array = []
    for item in array:
        new_array.extend([item] * n)
    new_array = np.array(new_array)
    return new_array


print(expandArray([1,2,3,4], 3)

[1,1,1,2,2,2,3,3,3,4,4,4]

Numpy's repeat, is exactly what you want numpy.org/doc/stable/reference/generated/numpy.repeat.html — Kevin S
– Kevin S, Commented Feb 26, 2021 at 0:18

Dániel Varga · Accepted Answer · 2021-02-26 00:33:18Z

1

I don't know exactly why, but this code runs faster than np.repeat for me:

def expandArray(array, n):
    return np.concatenate([array for i in range(0,n)])

I ran this little benchmark:

arr1 = np.random.rand(100000)
%timeit expandArray(arr1, 5)

1.07 ms ± 25.8 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)

And np.repeat gives this result:

%timeit np.repeat(arr1,5)

2.45 ms ± 148 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)

answered Feb 26, 2021 at 0:33

Dániel Varga

212 bronze badges

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

2 Comments

Thang Pham Over a year ago

Your solution produces a different array [1 2 3 4 1 2 3 4 1 2 3 4]. If their orders are not required then this is a good solution.

ProteinGuy Over a year ago

The order is very important.

ProteinGuy · Accepted Answer · 2022-05-04 20:30:31Z

0

You can create a duplicated array with the numpy.repeat function.

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

new_array = np.repeat(array, repeats=3, axis=None)
print(new_array)

array([1,1,1,2,2,2,3,3,3,4,4,4])


new_array = np.repeat(array.reshape(-1,1).transpose(), repeats=3, axis=0).flatten()
print(new_array)

array([1,2,3,4,1,2,3,4,1,2,3,4])

answered May 4, 2022 at 20:30

ProteinGuy

1,9622 gold badges22 silver badges39 bronze badges

1 Comment

James Mar 18 at 13:50

For the second part, to repeat the entire array, numpy.tile is available: new_array = np.tile(array, 3)

Collectives™ on Stack Overflow

Create new numpy array by duplicating each item in an array

2 Answers 2

2 Comments

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

2 Comments

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related