Jak uzyskać jedną liczbę konkretnych razy w python tablicy

Question

Mam jedną tablicę jak A = [1,2,3] i innej tablicy B = [4,5,6]. Teraz muszę kolejną tablicę C tak, że elementy w C powinny być te same elementy B o wystąpieniu w kolejności elementem A. podobnych, 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 

Wynik:

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

ten jest to odpowiednik jednej linii do:

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

... co jest z grubsza odpowiednikiem

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. Nie daj się zaskoczyć przez podkreślenie. It's an ordinary variable. To jest zwykle używany, gdy w rzeczywistości nie muszą odnosić się do wartości zmiennej, tak jak w tym przykładzie)

Answer 2

Moje rozwiązanie:

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

wyjaśnieniu:

[B[i]]*A[i] stworzy listę A[i] pozycje B[i]

[B[i]]*A[i] for i in range(len(A))] daje listę list np [[4], [5, 5], [6, 6, 6]]

sum(C, []) przekonwertuje listę list do listy

Answer 3

podam inny sposób to zrobić, stosując inny pomysł (nie twierdząc, powinien to wykorzystać, ale ze względów dydaktycznych)

znajdę dość schludny być zdolne do replikacji elementy w Pythonie przy użyciu

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

jeśli ja po prostu to zrobić:

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

otrzymuję grupy elementów:

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

Następnie możemy użyć itertools.chain wrócić do jednej listy:

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

Answer 4

Oto kolejny rozwiązanie za pomocą numpy:

import numpy 

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

c = numpy.repeat(b,a) 

Wynik:

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

Tutaj Więcej informacji o używaniu numpy.repeat: http://docs.scipy.org/doc/numpy/reference/generated/numpy.repeat.html

Answer 5

Innym sposobem:

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

Ale wolałbym listę zrozumieniem, tylko bez uciążliwego _item, tj,

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

Lub krócej (ale dla dużych przypadkach wolno) Suma:

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