计算嵌套列表某一层次的元素数量<新人求助>(可以的话麻烦看看我的代码怎么改)
【问题描述】输入一个嵌套列表,再输入层数,求该层的元素个数。如果求第二层次的元素个数,但是第二层次还有第三层次嵌套,则嵌套列表算一个元素。
【输入形式】
第一行输入嵌套列表,包含方括号,逗号分隔,每个元素是整数
第二行输入需要计算的层次,整数,大于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 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]