从python的另一个列表中创建多个副本列表

Question

numpy中的任何函数都可以实现这个功能吗？以下功能f是一种尴尬

def f(l,times): 
    res=[] 
    for i in range(len(l)): 
     res+=[l[i]]*times[i] 
    return res 



In [93]:f([1,2,3],[2,2,2]) 

Out [93]:[1, 1, 2, 2, 3, 3] 

Answer 1

np.repeat正是这样做的。 例：

In [8]: a = np.arange(4) 

In [9]: b = np.array([1, 2, 1, 3]) 

In [10]: np.repeat(a, b) 
Out[10]: array([0, 1, 1, 2, 3, 3, 3]) 

如果使用> = 2个维阵列时，可以指定一个轴线参数。请参阅文档here。