Python 输入一个嵌套列表nums,再输入一个层数n,求该层的数字元素个数。
时间: 2024-02-16 08:32:22 浏览: 77
题库:计算嵌套列表某一层次的元素数量.py
可以使用递归函数来实现该功能。首先判断当前列表是否为一个数字列表,如果是,则返回该列表的长度,否则递归遍历其子列表,层数减1,直到遍历到第n层时返回该层的数字元素个数。
代码如下:
```python
def count_elements(nums, n):
if n == 1: # 遍历到第n层
return len(nums)
count = 0
for item in nums:
if isinstance(item, int): # 判断是否为数字
count += 1
elif isinstance(item, list): # 判断是否为列表
count += count_elements(item, n-1)
return count
# 示例
nums = [1, [2, 3], [4, [5, 6]], [[7], 8]]
print(count_elements(nums, 1)) # 输出:1
print(count_elements(nums, 2)) # 输出:3
print(count_elements(nums, 3)) # 输出:4
print(count_elements(nums, 4)) # 输出:1
```
阅读全文