[已解决]计算嵌套列表中某一层列表的数字元素个数<新人求助>

akcw

【问题描述】

输入一个嵌套列表nums，再输入一个层数n，求该层的数字元素个数。
【输入形式】

第一行输入列表，元素是自然数，包括方括号，逗号分隔
【输出形式】

整数
【样例输入】

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

3

【样例输出】

2
【样例说明】

输入样例的嵌套有三个层次，当输入3的时候，就是计算第三层次嵌套列表的数字元素个数。 如果输入的是2，则第2层次嵌套列表中，数字元素的个数是3.

我的代码在测试时有两个数据不通过，但我自己执行的时候是可以的，希望能看到大佬的代码对比然后改进自己的代码

最佳答案

月排行榜 / 总排行榜

ba21

2021-5-22 15:28:25

qq1151985918 发表于 2021-5-22 14:10
刚看了下，是不太一样，上个帖子 嵌套列表算一个元素，这个帖子不算，不过也没难多少，改一行代码就好了 ...

你2个贴的代码，和我对题目理解的不一至。不知道是不是我想多了。于是我上网找了下该题的相关资料，如下：

1. data = [1, 2, [3, 4, [5, 6], 7, [5, 6]], 8, [3, 4, [5, 6], 7]]
   
2. n = 3
   
3. 
   
4. # 该函数是我以为的要达到的
   
5. # 来自 https://blog.csdn.net/liurongke/article/details/116305467
   
6. def function(lis,lev,rank):
   
7.     cnt = 0
   
8.     for i in lis:
   
9.         if rank != lev and isinstance(i,list):    #如果当前层数不是指定层数，并且该元素为列表，那么我们就进入该列表，然后增加一个层数
   
10.             cnt += function(i,lev,rank+1)
    
11.         elif rank == lev and isinstance(i,int) or isinstance(i,float):   #如果是当前层数，并且是数字，就加起来
    
12.             cnt += 1
    
13.     return cnt
    
14. 
    
15. print(function(data,n,1))               #函数（列表，指定层数，从第一层开始）
    
16. 
    
17. 
    
18. # 该函数也是我以为的要达到的
    
19. # 来自 https://www.cnblogs.com/dreamcoding/p/12694572.html
    
20. def getSum(items,depth,n):
    
21.     if type(items) == int:
    
22.         if type(items) == int:
    
23.             if depth == n:
    
24.                 return 1
    
25.             else:
    
26.                 return 0
    
27.     else:
    
28.         if (type(items) == tuple) or (type(items) == list):
    
29.             result = 0
    
30.             for i in range(len(items)):
    
31.                 result += getSum(items[i],depth+1,n)
    
32.             return result
    
33.         else:
    
34.             return 0
    
35. result = 0
    
36. for i in range(len(data)):
    
37.     result = result + getSum(data[i],1,n)
    
38. print(result)
    
39. 
    
40. 
    
41. 
    
42. # 该函数也是我以为的要达到的 (实现方式比效另类, 效率明显要比递归快)
    
43. # 来自 https://tuenity.blog.csdn.net/article/details/102722592
    
44. a = '[1, 2, [3, 4, [5, 6], 7, [5, 6]], 8, [3, 4, [5, 6], 7]]'
    
45. c = n
    
46. b = ''.join(a)
    
47. num= 0
    
48. sum=0
    
49. for i in range(0,len(b)):
    
50.     if(b[i]=='['):
    
51.         num+=1
    
52.     elif(b[i]==']'):
    
53.         num-=1
    
54.     if (c == num) and (b[i]!='[') and (b[i]!=']' and b[i]!=',' and b[i+1].isdigit()==False):
    
55.         sum += 1
    
56. print(sum)
    
57. 
    
58. 
    
59. # 然后就是你们认为的结果
    
60. def level(data:list) -> list:
    
61.     res = [len([i for i in data if type(i) == int])]
    
62.     for i in data:
    
63.         if type(i) == list:
    
64.             res += level(i)
    
65.     return res
    
66. 
    
67. print(level(data)[num - 1])
    
68. 
    

复制代码

跳转到最佳答案楼层

qq1151985918

<font color="red">[已解决]</font>计算嵌套列表某一层次的元素数量&lt;新人求助&gt;（可以的话麻烦看看我的代码怎么改）
https://fishc.com.cn/thread-196130-1-1.html
(出处: 鱼C论坛)

刚发现本题跟我所回复的帖子中有所不同，上一个嵌套列表算一个元素，这一个是计算数字数量，有所区别，请对照。

1. def level(data:list) -> list:
   
2.     res = [len([i for i in data if type(i) == int])]
   
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])
    

复制代码

ba21

qq1151985918 发表于 2021-5-22 13:19
[已解决]计算嵌套列表某一层次的元素数量&lt;新人求助&gt;（可以的话麻烦看看我的代码怎么改）
https://fish ...

求该层的数字元素个数

难度比你帖子上的难度大多了

qq1151985918

ba21 发表于 2021-5-22 13:59
求该层的数字元素个数

难度比你帖子上的难度大多了

刚看了下，是不太一样，上个帖子 嵌套列表算一个元素，这个帖子不算，不过也没难多少，改一行代码就好了。

1. def level(data:list) -> list:
   
2.     res = [len([i for i in data if type(i) == int])]
   
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

我刚注意到，原来这两个帖子都是你问的。

ba21

qq1151985918 发表于 2021-5-22 14:10
刚看了下，是不太一样，上个帖子 嵌套列表算一个元素，这个帖子不算，不过也没难多少，改一行代码就好了 ...

你2个贴的代码，和我对题目理解的不一至。不知道是不是我想多了。于是我上网找了下该题的相关资料，如下：

1. data = [1, 2, [3, 4, [5, 6], 7, [5, 6]], 8, [3, 4, [5, 6], 7]]
   
2. n = 3
   
3. 
   
4. # 该函数是我以为的要达到的
   
5. # 来自 https://blog.csdn.net/liurongke/article/details/116305467
   
6. def function(lis,lev,rank):
   
7.     cnt = 0
   
8.     for i in lis:
   
9.         if rank != lev and isinstance(i,list):    #如果当前层数不是指定层数，并且该元素为列表，那么我们就进入该列表，然后增加一个层数
   
10.             cnt += function(i,lev,rank+1)
    
11.         elif rank == lev and isinstance(i,int) or isinstance(i,float):   #如果是当前层数，并且是数字，就加起来
    
12.             cnt += 1
    
13.     return cnt
    
14. 
    
15. print(function(data,n,1))               #函数（列表，指定层数，从第一层开始）
    
16. 
    
17. 
    
18. # 该函数也是我以为的要达到的
    
19. # 来自 https://www.cnblogs.com/dreamcoding/p/12694572.html
    
20. def getSum(items,depth,n):
    
21.     if type(items) == int:
    
22.         if type(items) == int:
    
23.             if depth == n:
    
24.                 return 1
    
25.             else:
    
26.                 return 0
    
27.     else:
    
28.         if (type(items) == tuple) or (type(items) == list):
    
29.             result = 0
    
30.             for i in range(len(items)):
    
31.                 result += getSum(items[i],depth+1,n)
    
32.             return result
    
33.         else:
    
34.             return 0
    
35. result = 0
    
36. for i in range(len(data)):
    
37.     result = result + getSum(data[i],1,n)
    
38. print(result)
    
39. 
    
40. 
    
41. 
    
42. # 该函数也是我以为的要达到的 (实现方式比效另类, 效率明显要比递归快)
    
43. # 来自 https://tuenity.blog.csdn.net/article/details/102722592
    
44. a = '[1, 2, [3, 4, [5, 6], 7, [5, 6]], 8, [3, 4, [5, 6], 7]]'
    
45. c = n
    
46. b = ''.join(a)
    
47. num= 0
    
48. sum=0
    
49. for i in range(0,len(b)):
    
50.     if(b[i]=='['):
    
51.         num+=1
    
52.     elif(b[i]==']'):
    
53.         num-=1
    
54.     if (c == num) and (b[i]!='[') and (b[i]!=']' and b[i]!=',' and b[i+1].isdigit()==False):
    
55.         sum += 1
    
56. print(sum)
    
57. 
    
58. 
    
59. # 然后就是你们认为的结果
    
60. def level(data:list) -> list:
    
61.     res = [len([i for i in data if type(i) == int])]
    
62.     for i in data:
    
63.         if type(i) == list:
    
64.             res += level(i)
    
65.     return res
    
66. 
    
67. print(level(data)[num - 1])
    
68. 
    

复制代码

qq1151985918

ba21 发表于 2021-5-22 15:28
你2个贴的代码，和我对题目理解的不一至。不知道是不是我想多了。于是我上网找了下该题的相关资料，如下 ...

哦，你是认为如果同一层次可能有多个列表，可能题目样例没体现出来，其实那样用我这个递归也并不难

1. data = [1, 2, [3, 4, [5, 6], 7, [5, 6]], 8, [3, 4, [5, 6], 7]]
   
2. n = 3
   
3. def level(data:list) -> list:
   
4.     res = [len([i for i in data if type(i) == int])]
   
5.     same_level = []
   
6.     for i in data:
   
7.         if type(i) == list:
   
8.             same_level += i
   
9.     if same_level:
   
10.         res += level(same_level)
    
11.     return res
    
12. 
    
13. print(level(data)[n - 1])
    

复制代码

ba21

qq1151985918 发表于 2021-5-22 15:40
哦，你是认为如果同一层次可能有多个列表，可能题目样例没体现出来，其实那样用我这个递归也并不难

不错
不过还是另类那个好。。

qq1151985918

ba21 发表于 2021-5-22 15:48
不错
不过还是另类那个好。。

另类那个比较容易出bug，比如说，如果输入了两个空格就会出bug，如果有负号也会有bug，当然负数不是自然数，
或许改成  b = str(eval(a)) 好一些
或者把最后一个条件改成 (b[ i ].isdigit()==True and b[i+1].isdigit()==False)

akcw

感谢，我看看