python - 根据另一个列表中的值重复的连续数字列表

Question

我的目标是获得一个连续数字的列表，并根据初始列表值相应地重复。假设我有:

initialList=[1,2,3,5]

我想得到:

targetList=[0,1,1,2,2,2,3,3,3,3,3]

...我是 Python 的新手，对于这个 - 可能 - 非常第一步的问题，我深表歉意。不幸的是，尝试了很多搜索，但结果与我的需求不符。非常感谢您。

Answer 1

新手友好的解决方案是使用两个循环:

result = []
number = 0
for repeat in initialList:
    for _ in range(repeat):
        result.append(number)
    number += 1

print(result)  # [0, 1, 1, 2, 2, 2, 3, 3, 3, 3, 3]

---

如果出于任何原因您更喜欢单行，您可以结合 enumerate和 range得到

result = [num for num, repeat in enumerate(initialList) for _ in range(repeat)]