Cách lấy một số lần cụ thể trong một mã vạch trăn

Question

Tôi có một mảng như A = [1,2,3] và một mảng khác B = [4,5,6]. Bây giờ, tôi cần một mảng C để các yếu tố trong C nên các yếu tố tương tự của B có xảy ra theo thứ tự của nguyên tố A. Giống như, 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 

Kết quả:

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

này là một trong những dòng tương đương với:

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

... Đó là tương đương với

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. Đừng bị vấp ngã bởi dấu gạch dưới. It's an ordinary variable. Người ta thông thường được sử dụng khi bạn không thực sự phải tham khảo giá trị của biến, như trong ví dụ này)

Answer 2

Giải pháp của tôi:

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

Giải thích:

[B[i]]*A[i] sẽ tạo ra một danh sách với A[i] hạng mục công B[i]

[B[i]]*A[i] for i in range(len(A))] sẽ cung cấp cho bạn một danh sách liệt kê, ví dụ như [[4], [5, 5], [6, 6, 6]]

sum(C, []) sẽ chuyển đổi danh sách liệt kê vào danh sách

Answer 3

tôi sẽ cung cấp một cách khác để làm việc đó, sử dụng một ý tưởng khác nhau (không tuyên bố ông nên sử dụng điều này, nhưng vì lý do giáo khoa)

tôi thấy khá gọn gàng để được có khả năng tái tạo các yếu tố trong python sử dụng

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

nếu tôi chỉ đơn giản là làm điều này:

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

tôi nhận được nhóm yếu tố:

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

Sau đó, chúng ta có thể sử dụng itertools.chain để có được trở lại với một danh sách duy nhất:

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

Answer 4

Đây là một giải pháp sử dụng NumPy:

import numpy 

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

c = numpy.repeat(b,a) 

Kết quả:

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

Đây biết thêm thông tin về việc sử dụng numpy.repeat: http://docs.scipy.org/doc/numpy/reference/generated/numpy.repeat.html

Answer 5

Một cách khác:

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

Nhưng tôi muốn danh sách hiểu, chỉ nếu không có sự gây phiền nhiễu _item, ví dụ:,

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

Hoặc ngắn hơn (nhưng đối với trường hợp lớn chậm) tổng:

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