python - 重复整数列表

Question

创建给定大小的列表的最 pythonic 方法是什么 N= l*k 其中 l 是不同符号的数量(为简单起见，为整数)和 k 是这样的子序列长度:

N=12, l=4, k=3

[ 0,0,0, 1,1,1, 2,2,2, 3,3,3 ]

或者例如 N=15 l=3, k=5:

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

这个函数应该经常被调用，所以速度是可取的。

Answer 1

使用 numpy 你可以这样做:

In [23]: import numpy as np

In [26]: a=np.arange(3).repeat(5)

In [27]: a
Out[27]: array([0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2])

或 python 内置:

In [29]: [l for l in range(3) for k in range(5)]
Out[29]: [0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2]