A = [1, [2, 3], [4, 5, 6], 7] B = [2, 3, 4, 5, 6, 7, 8] How to transform B into the same structure as A? E.g. B' = [2, [3, 4], [5, 6, 7], 8]

transform.py

def func(l1, l2):
    def traverse(l1, l2, i):
        res = []
        for e in l1:
            if isinstance(e, list):
                res.append(traverse(e, l2, i))
            else:
                res.append(l2[i[0]])
                i[0] += 1
        return res
    return traverse(l1, l2, [0])