numpy.repeat() in Python
Last Updated :
28 Mar, 2022
Improve
The numpy.repeat() function repeats elements of the array - arr. Syntax :
numpy.repeat(arr, repetitions, axis = None)
Parameters :
array : [array_like]Input array.
repetitions : No. of repetitions of each array elements along the given axis.
axis : Axis along which we want to repeat values. By default, it returns
a flat output array.
Return :
An array with repetitions of array - arr elements as per repetitions, number of times we want to repeat arr
Code 1 :
# Python Program illustrating
# numpy.repeat()
import numpy as geek
#Working on 1D
arr = geek.arange(5)
print("arr : \n", arr)
repetitions = 2
a = geek.repeat(arr, repetitions)
print("\nRepeating arr 2 times : \n", a)
print("Shape : ", a.shape)
repetitions = 3
a = geek.repeat(arr, repetitions)
print("\nRepeating arr 3 times : \n", a)
# [0 0 0 ..., 4 4 4] means [0 0 0 1 1 1 2 2 2 3 3 3 4 4 4]
# since it was long output, so it uses [ ... ]
print("Shape : ", a.shape)
Output :
arr : [0 1 2 3 4] Repeating arr 2 times : [0 0 1 1 2 2 3 3 4 4] Shape : (10,) Repeating arr 3 times : [0 0 0 ..., 4 4 4] Shape : (15,)
Code 2 :
# Python Program illustrating
# numpy.repeat()
import numpy as geek
arr = geek.arange(6).reshape(2, 3)
print("arr : \n", arr)
repetitions = 2
print("\nRepeating arr : \n", geek.repeat(arr, repetitions, 1))
print("arr Shape : \n", geek.repeat(arr, repetitions).shape)
repetitions = 2
print("\nRepeating arr : \n", geek.repeat(arr, repetitions, 0))
print("arr Shape : \n", geek.repeat(arr, repetitions).shape)
repetitions = 3
print("\nRepeating arr : \n", geek.repeat(arr, repetitions, 1))
print("arr Shape : \n", geek.repeat(arr, repetitions).shape)
Output :
arr : [[0 1 2] [3 4 5]] Repeating arr : [[0 0 1 1 2 2] [3 3 4 4 5 5]] arr Shape : (12,) Repeating arr : [[0 1 2] [0 1 2] [3 4 5] [3 4 5]] arr Shape : (12,) Repeating arr : [[0 0 0 ..., 2 2 2] [3 3 3 ..., 5 5 5]] arr Shape : (18,)
References : https://docs.scipy.org/doc/numpy/reference/generated/numpy.repeat.html Note : These codes won’t run on online IDE's. Please run them on your systems to explore the working .
