当前位置:首页python > 正文

Python如何求多维数组的元素个数

作者:野牛程序员:2023-07-14 07:56:01python阅读 2642

求多维数组的元素个数,可以通过递归的方式来实现。下面是一个示例的递归函数,用于计算多维数组的元素个数:

def count_elements(arr):
    count = 0
    if isinstance(arr, list):
        for item in arr:
            count += count_elements(item)
    else:
        count = 1
    return count

这个函数接受一个多维数组 arr 作为输入,并返回数组中元素的总个数。它通过递归地遍历数组的每个元素,如果遇到子数组,则对子数组进行递归调用,将子数组中的元素个数累加到总个数中。如果遇到的是非数组元素,则将计数器加1。

以下是一个示例用法:

# 定义一个多维数组
my_array = [1, 2, [3, 4, [5, 6], 7], 8, [9, [10, 11]]]

# 调用 count_elements 函数计算元素个数
element_count = count_elements(my_array)

# 打印结果
print("多维数组的元素个数为:", element_count)

输出:

多维数组的元素个数为: 11

在这个示例中,多维数组 my_array 包含11个元素,包括整数和子数组。通过调用 count_elements 函数,我们可以计算出多维数组的元素个数为11。

野牛程序员教少儿编程与信息学奥赛-微信|电话:15892516892
野牛程序员教少儿编程与信息学竞赛-微信|电话:15892516892
相关推荐

最新推荐

热门点击