Объединить массив с собой n раз

Question

Учитывая массив [3, 2, 1] формы (3,), я хотел бы расширить его до двумерного массива (n, 3), повторяя массив вниз, поэтому, где n = 3, это будет:

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

Какой самый простой способ сделать это?

(Другой способ выразить это: я могу создать n копий списка с помощью li * n. Есть ли способ сделать это с массивами? Если так, Я мог бы использовать это как ввод и изменить форму на (n,3))

Answer 1

Мы можем сделать это, используя np.repeat :

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

x = x.reshape(1,3) # replace 3 with num elems in your array

np.repeat(a=x, repeats=3, axis=0) # replace 3 with number of repeats you want

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

Надеюсь, это помогло ..