Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Mở rộng danh sách theo K trong Python


Mở rộng list bằng cách lặp lại mỗi phần tử K lần.

Cách 1: Vòng lặp + toán tử nhân list.

numbers = [1, 2, 3]
K = 5
result = []
for i in numbers:
    result += [i] * K
print(result)  # [1,1,1,1,1, 2,2,2,2,2, 3,3,3,3,3]

Cách 2: List comprehension lồng.

numbers = [1, 2, 3]
K = 5
result = [i for i in numbers for _ in range(K)]
print(result)  # [1,1,1,1,1, 2,2,2,2,2, 3,3,3,3,3]

Kết luận

Cách 2 ngắn gọn, Pythonic.