Python Numpy重复一个arange数组

时间:2016-12-21 02:51:33

标签: arrays numpy repeat numpy-ndarray

所以说我这样做

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])

3 个答案:

答案 0 :(得分:4)

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

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])

答案 1 :(得分:4)

我最近看过几个关于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])

答案 2 :(得分:2)

这个怎么样?

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

<强>输出

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

我不会说太多开销。