從另一個列表中

Question

我想從零級有關的數字從創建重複的連續編號列表創建關於數字重複的連續號碼列表list1

list1 = [3,5,4,1,2,2] 

預計名單應該有3個零，5分的人，4個三三兩兩， 等等。

list2 = [0,0,0,1,1,1,1,1,2,2,2,2,3,4,4,5,5] 

Answer 1

可以使用range，使重複使用嵌套列表理解：

lst = [i for i in range(len(list1)) for _ in range(list1[i])] 
print lst 
# [0, 0, 0, 1, 1, 1, 1, 1, 2, 2, 2, 2, 3, 4, 4, 5, 5] 

而且有更直觀的itertools方法：

from itertools import repeat, count 

c = count() 
lst = [x for i in list1 for x in repeat(next(c), i)] 

Answer 2

你可以做到這一點列表理解，但使用傳統的for循環和.extend方法。

list1 = [3,5,4,1,2,2] 
list2 = [] 
for i, v in enumerate(list1): 
    list2.extend([i] * v) 
print(list2) 

輸出

[0, 0, 0, 1, 1, 1, 1, 1, 2, 2, 2, 2, 3, 4, 4, 5, 5] 

---

另一種選擇是使用itertools.chain：

list2 = list(chain.from_iterable([i] * v for i, v in enumerate(list1))) 

，或者在略短，但更隱蔽

list2 = list(chain(*([i] * v for i, v in enumerate(list1)))) 

Answer 3

你可以使用一個列表的理解和enumerate()遍歷計數/值對：

>>> list1 = [3, 5, 4, 1, 2, 2] 
>>> [j for i, n in enumerate(list1) for j in [i]*n] 
[0, 0, 0, 1, 1, 1, 1, 1, 2, 2, 2, 2, 3, 4, 4, 5, 5] 

或者，你可以使用itertools.chain（或chain.from_iterable），特別是如果你並不總是需要建立一個列表在內存中：

>>> from itertools import chain 
>>> list(chain(*[[i]*n for i, n in enumerate(list1)])) 
[0, 0, 0, 1, 1, 1, 1, 1, 2, 2, 2, 2, 3, 4, 4, 5, 5]