Wie bekomme ich eine bestimmte Anzahl von Zeiten in einem Array Python

Question

Ich habe ein Array wie A = [1,2,3] und ein anderes Array B = [4,5,6]. Nun muss ich ein weiteres Array C so, dass die Elemente in C in der Größenordnung des Elements A. Wie die gleichen Elemente von B mit Auftreten sein sollte, C = [4, 5, 5, 6, 6, 6]

Answer 1

A = [1,2,3] 
B = [4,5,6] 
C = [b_item for a_item, b_item in zip(A,B) for _ in range(a_item)] 
print C 

Ergebnis:

[4, 5, 5, 6, 6, 6] 

Diese ist eine einzeilige entspricht:

C = [] 
for a_item, b_item in zip(A,B): 
    for _ in range(a_item): 
     C.append(b_item) 

... was in etwa gleichwertig

ist

C = [] 
for i in range(min(len(A), len(B))): 
    a_item = A[i] 
    b_item = B[i] 
    for _ in range(a_item): 
     C.append(b_item) 

(N.B. Stolpern Sie nicht über den Unterstrich. It's an ordinary variable. Es wird üblicherweise verwendet, wenn Sie tatsächlich nicht auf den Wert der Variablen, wie in diesem Beispiel beziehen müssen)

Answer 2

Meine Lösung:

C = sum([[B[i]]*A[i] for i in range(len(A))], []) 
print C 

Explaination:

[B[i]]*A[i] eine Liste erstellen mit A[i] Artikel von B[i]

[B[i]]*A[i] for i in range(len(A))] finden Sie eine Liste von Listen, zB [[4], [5, 5], [6, 6, 6]]

012 geben

sum(C, []) werden Liste der Listen in Liste konvertieren

Answer 3

ich einen anderen Weg, es zu tun, mit einer anderen Idee liefern wird (nicht behaupten, er diese nutzen sollte, aber aus didaktischen Gründen)

finde ich recht ordentlich zu sein mit der Lage Elemente in python zu replizieren

[0]*3 # it means we create 3 elements of zero: [0,0,0] 

wenn ich diese einfach tun:

[[x]*i for i,x in zip(A, B)] 

ich Gruppen von Elementen:

[[4], [5, 5], [6, 6, 6]] 

Dann könnten wir itertools.chain, um wieder zu einer einzigen Liste verwenden:

from itertools import chain 
list(chain(*[[x]*i for i,x in zip(A, B)])) 
[4, 5, 5, 6, 6, 6] 

Answer 4

Hier ist eine andere Lösung numpy mit:

import numpy 

a = numpy.array = [1,2,3] 
b = numpy.array = [4,5,6] 

c = numpy.repeat(b,a) 

Ergebnis:

array([4, 5, 5, 6, 6, 6]) 

Hier Weitere Informationen über numpy.repeat mit: http://docs.scipy.org/doc/numpy/reference/generated/numpy.repeat.html

Answer 5

Einen anderen Weg:

C = [] 
for a, b in zip(A, B): 
    C.extend([b] * a) 

Aber ich die Liste Verständnis bevorzugen würde, nur ohne die lästigen _item, das heißt,

C = [b for a, b in zip(A, B) for _ in range(a)] 

oder die kürzere (aber für große Fälle langsam) Summe:

C = sum(([b] * a for a, b in zip(A, B)), [])