嵌套列表的问题变动了一下应该怎么做呢？

输入一个嵌套列表，再输入层数，求该层的元素个数。如果求第二层次的元素个数，但是第二层次还有第三层次嵌套，则嵌套列表算一个元素。

def cp(li,n):
    t = li
    for i in range(n-1):
        g = []
        for v in t:
            if type(v) is list:
                g.extend(v)
        t = g

    return len(t)

lis = [
    [
        [1,0,1,0],
        [1,0,1,0],
        [1,0,1,0]
    ],
    [
        [1,0,1,0],
        [1,0,1,0],
        [1,0,1,0]
    ]
]

print(cp(lis,1))
print(cp(lis,2))
print(cp(lis,3))6

您好，我是有问必答小助手，您的问题已经有小伙伴解答了，您看下是否解决，可以追评进行沟通哦~

如果有您比较满意的答案 / 帮您提供解决思路的答案，可以点击【采纳】按钮，给回答的小伙伴一些鼓励哦～～

ps:问答VIP仅需29元，即可享受5次/月 有问必答服务，了解详情>>>https://vip.csdn.net/askvip?utm_source=1146287632