Python求指定层的元素个数

输入一个嵌套列表或元组，再输入层数，求该层的数字元素个数。

输入格式:

第一行输入列表或元组
第二行输入层数

输出格式:

在一行中输出元素个数

输入样例:

在这里给出一组输入。例如：

[1,2,[3,4,[5,6],7],8]
3

输出样例:

在这里给出相应的输出。例如：

2

#数据分析求定层的元素个数
def loose (cou,x,weight,floor):
    #进阶判断
    if type(x) == list or type(x) == tuple:
        weight += 1
        #计数判断
        for y in x:
            if weight == floor and type(y) != list and type(y) != tuple:
                    cou += 1
            else:
                cou = loose(cou,y,weight,floor)
        weight -= 1
        return cou
    else:
        return cou
#定义列表并eval处理
List = eval(input())
#定义目标层
floor = int(input())
#定义层数变量
weight = 1
#声明计数器
cou = 0
for x in List:
    cou = loose(cou,x,weight,floor)
print(cou)

以上代码全为本人亲自手敲，可能有一些错误和不足之处，如有更好的方法和建议，欢迎您在评论区友善讨论。