Функция repeat

Функция repeat повторяет каждый элемент массива указанное количество раз. Первым параметром функция принимает массив, вторым - количество повторений для каждого элемента. Третьим параметром можно задать ось, вдоль которой выполняется повторение.

Синтаксис

np.repeat(arr, repeats, [axis])

Пример

Давайте повторим каждый элемент одномерного массива два раза:

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
res = np.repeat(arr, 2)
print(res)

Результат выполнения кода:

[1 1 2 2 3 3 4 4 5 5]

Пример

Давайте повторим каждый элемент одномерного массива разное количество раз:

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
repeats = [2, 1, 3, 2, 1]
res = np.repeat(arr, repeats)
print(res)

Результат выполнения кода:

[1 1 2 3 3 3 4 4 5]

Пример

Давайте повторим элементы двумерного массива по строкам:

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
res = np.repeat(arr, 2, axis=0)
print(res)

Результат выполнения кода:

[[1 2 3]
 [1 2 3]
 [4 5 6]
 [4 5 6]]

Пример

Давайте повторим элементы двумерного массива по столбцам:

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
res = np.repeat(arr, 2, axis=1)
print(res)

Результат выполнения кода:

[[1 1 2 2 3 3]
 [4 4 5 5 6 6]]

Пример

Давайте повторим каждый элемент двумерного массива разное количество раз по строкам:

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
repeats = [2, 1]
res = np.repeat(arr, repeats, axis=0)
print(res)

Результат выполнения кода:

[[1 2 3]
 [1 2 3]
 [4 5 6]]

Пример

Давайте повторим каждый элемент двумерного массива разное количество раз по столбцам:

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
repeats = [1, 2, 1]
res = np.repeat(arr, repeats, axis=1)
print(res)

Результат выполнения кода:

[[1 2 2 3]
 [4 5 5 6]]

Пример

Давайте повторим элементы строкового массива три раза:

import numpy as np

arr = np.array(['a', 'b', 'c', 'd', 'e'])
res = np.repeat(arr, 3)
print(res)

Результат выполнения кода:

['a' 'a' 'a' 'b' 'b' 'b' 'c' 'c' 'c' 'd' 'd' 'd' 'e' 'e' 'e']

Смотрите также

функцию tile,
которая повторяет весь массив заданное количество раз
функцию reshape,
которая изменяет форму массива без изменения данных
функцию concatenate,
которая объединяет несколько массивов вдоль существующей оси
функцию stack,
которая объединяет массивы вдоль новой оси