python 如何组成这个格式

Question

['1', '1', '1', '1', '2', '2', '2', '3', '3', '4', '4', '5']这里是有规律的,从1排到x我要组成0:4 4:7 7:9 9:11 11:120:4 就是从0 开始4个, 4:7 就是 从4开始到最后一个2 7:9 开始就是从7开始到最后一个3 :依次类推,组成一个字典或list

Answer 1

看起来 0:4, 4:7，... 可视为数组的下标，可简化成元素个数 4, 3，...

那么问题演变成“由同元素个数组成的序列，生成相应的数组”，如下

def compute(*size_list):
    l2 = []
    for i, size in enumerate(size_list):
        l2.extend([str(i+1)] * size)
    return l2


def test_compute():
    assert compute(4, 3, 2, 2, 1) == [
        '1', '1', '1', '1',
        '2', '2', '2',
        '3', '3',
        '4', '4',
        '5',
    ]