Liste mehrerer Kopien aus einer anderen Liste in Python erstellen

Question

Jede Funktion in numpy kann dies erreichen? Funktion f unten ist eine Art umständlich

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 tut genau dies. Ex:

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

Wenn Sie mit> = 2-dimensionalen Arrays arbeiten, können Sie eine Achse Parameter angeben. Siehe das Dokument here.