Python 如何使用 np tile reshape 数组

发布于03月27日

我有一个X数组，形状为2，5，如下所示:

0, 6, 7, 9, 1
2, 4, 6, 2, 7

我想对其进行reshape ，使其每行重复n次，如下所示(示例使用n=3):

0, 6, 7, 9, 1
0, 6, 7, 9, 1
0, 6, 7, 9, 1
2, 4, 6, 2, 7
2, 4, 6, 2, 7
2, 4, 6, 2, 7

我试过使用np.平铺如下，但重复如下所示:

np.tile(X, (3, 5))
0, 6, 7, 9, 1
2, 4, 6, 2, 7
0, 6, 7, 9, 1
2, 4, 6, 2, 7
0, 6, 7, 9, 1
2, 4, 6, 2, 7

如何有效地创建所需的输出？

推荐答案

如果a是主数组:

a = np.array([0, 6, 7, 9, 1, 2, 4, 6, 2, 7])

我们可以先将形状reshape 为所需的数组形状，然后使用np.repeat作为:

b = a.reshape(2, 5)
final = np.repeat(b, 3, axis=0)

它也可以用np.tile完成，但它需要不必要的额外操作，如下所示.因此，np.repeat将是更好的 Select .

test = np.tile(b, (3, 1))
final = np.concatenate((test[::2], test[1::2]))

实用课程推荐