akcw 发表于 2021-5-16 21:41:56

计算嵌套列表某一层次的元素数量<新人求助>(可以的话麻烦看看我的代码怎么改)

【问题描述】

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

【输入形式】

第一行输入嵌套列表,包含方括号,逗号分隔,每个元素是整数

第二行输入需要计算的层次,整数,大于0
【输出形式】

整数,计算所得的个数
【样例输入】

,7],8]

2

【样例输出】

4
【样例说明】

在第二层次 有3,4,, 7 等4个元素


list_lst=eval(input())
j=int(input())
lst=list_lst
lst1=
k=len(lst1)
print(k)

Traceback (most recent call last):
File "/85200050/p85200050", line 5, in <module>
    lst1=
TypeError: 'int' object is not iterable

qq1151985918 发表于 2021-5-16 22:40:48

def level(data:list) -> list:
    res =
    for i in data:
      if type(i) == list:
            res += level(i)
    return res

##data = , 7], 8]
data = eval(input("输入嵌套列表:"))
num = int(input("输入需要计算的层次:"))
print(level(data))
页: [1]
查看完整版本: 计算嵌套列表某一层次的元素数量<新人求助>(可以的话麻烦看看我的代码怎么改)