时间戳

首页 > 解决方案 > 重复二维数组的行

python - 重复二维数组的行

问题描述

我有一个 numpy 数组，我想重复它 n 次，同时保留行的原始顺序：

>>>a
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])

所需输出（对于 n =2）：

>>>a
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11],
       [ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])

我找到了一个 np.repeat 函数，但是，它不保留列的原始顺序。是否有任何其他内置函数或技巧可以在保留顺序的同时重复数组？

标签： pythonnumpy

解决方案

使用np.repeat，后跟np.concatenate：

np.concatenate(np.repeat(a[None, :], n, axis=0), axis=0)

array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11],
       [ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])

另一种选择是使用np.broadcast_to：

np.broadcast_to(a, (n, *a.shape)).reshape(-1, a.shape[1])

array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11],
       [ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])

推荐阅读