python - 如何将矩阵的每一行复制N次Numpy

Question

我有一个具有这些尺寸（150,2）的矩阵，我想将每一行复制 N 次。我用一个例子来说明我的意思。

输入：

a = [[2, 3], [5, 6], [7, 9]]

假设 N = 3，我想要这个输出：

[[2 3]
 [2 3]
 [2 3]
 [5 6]
 [5 6]
 [5 6]
 [7 9]
 [7 9]
 [7 9]]

谢谢你。

标签： pythonnumpymatrixduplicatesrow

Answer 1

np.repeat与参数一起使用axis=0：

a = np.array([[2, 3],[5, 6],[7, 9]])

print(a)
[[2 3]
 [5 6]
 [7 9]]

r_a = np.repeat(a, repeats=3, axis=0)

print(r_a)
[[2 3]
 [2 3]
 [2 3]
 [5 6]
 [5 6]
 [5 6]
 [7 9]
 [7 9]
 [7 9]]