Python Numpy重复排列数组

Question

所以说我这样做

x = np.arange(0, 3)

这使

array([0, 1, 2])

但是我能做什么

x = np.arange(0, 3)*repeat(N=3)times

要得到

array([0, 1, 2, 0, 1, 2, 0, 1, 2])

Answer 1

我最近看到了一些有关的问题resize。它不经常使用，但是在一种情况下，它可以满足您的要求：

In [66]: np.resize(np.arange(3),3*3)
Out[66]: array([0, 1, 2, 0, 1, 2, 0, 1, 2])

还有许多其他方式可以做到这一点。

In [67]: np.tile(np.arange(3),3)
Out[67]: array([0, 1, 2, 0, 1, 2, 0, 1, 2])
In [68]: (np.arange(3)+np.zeros((3,1),int)).ravel()
Out[68]: array([0, 1, 2, 0, 1, 2, 0, 1, 2])

np.repeat 不以我们想要的方式重复

In [70]: np.repeat(np.arange(3),3)
Out[70]: array([0, 0, 0, 1, 1, 1, 2, 2, 2])

但即使可以重做（这有点高级）：

In [73]: np.repeat(np.arange(3),3).reshape(3,3,order='F').ravel()
Out[73]: array([0, 1, 2, 0, 1, 2, 0, 1, 2])

Answer 2

编辑：请参阅 hpaulj 的答案。坦率地说，这是更好的。

最简单的方法是转换回列表并使用：

list(np.arange(0,3))*3

这使：

>> [0, 1, 2, 0, 1, 2, 0, 1, 2]

或者如果你想要它作为numpy数组：

np.array(list(np.arange(0,3))*3)

这使：

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

Answer 3

这个怎么样？

arr = np.arange(3)
res = np.hstack((arr, ) * 3)

输出

array([0, 1, 2, 0, 1, 2, 0, 1, 2])

我想说的开销并不大。