從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。