[已解决]计算嵌套列表某一层次的元素数量<新人求助>（可以的话麻烦看看我的代码怎么改）

akcw

【问题描述】

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

【输入形式】

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

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

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

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

2

【样例输出】

4
【样例说明】

在第二层次 有3,4，[5,6], 7 等4个元素


list_lst=eval(input())
j=int(input())
lst=list_lst[j]
lst1=[str(x) for x in lst]
k=len(lst1)
print(k)

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

最佳答案

月排行榜 / 总排行榜

qq1151985918

2021-5-16 22:40:48

1. def level(data:list) -> list:
   
2.     res = [len(data)]
   
3.     for i in data:
   
4.         if type(i) == list:
   
5.             res += level(i)
   
6.     return res
   
7. 
   
8. ##data = [1, 2, [3, 4, [5, 6], 7], 8]
   
9. data = eval(input("输入嵌套列表："))
   
10. num = int(input("输入需要计算的层次："))
    
11. print(level(data)[num - 1])
    

复制代码

跳转到最佳答案楼层

qq1151985918

1. def level(data:list) -> list:
   
2.     res = [len(data)]
   
3.     for i in data:
   
4.         if type(i) == list:
   
5.             res += level(i)
   
6.     return res
   
7. 
   
8. ##data = [1, 2, [3, 4, [5, 6], 7], 8]
   
9. data = eval(input("输入嵌套列表："))
   
10. num = int(input("输入需要计算的层次："))
    
11. print(level(data)[num - 1])
    

复制代码