如何修复此修改后的行程编码程序？

如何解决如何修复此修改后的行程编码程序？

嗨，我是python新手，我需要帮助

对于重复的连续字符，我想返回一个表示[len（x），x]（RLE）的嵌套列表。我创建了这个程序：

import itertools
lst = [1,1,2,3,4,5,1]
def function(lst):
    new = []
    for k,g in itertools.groupby(lst):
        if len(list(g)) > 1:
            new.append([len(list(g)),k])
        else:
            new.append(k)
    return new
print(function(lst))

但它似乎会返回：

[[0,1],[0,4],1]

为什么要这样做，我该如何解决？

谢谢！