获取列表列表的长度

Question

我有一些看起来有点奇怪的列表，如下所示：

listlist = [[], [[[[], []], [[]], []]], [[]]]

我试图找到所有列表长度的总和

如果它是一个简单的多维数组（nx nx nx…），那么查找列表长度的总和将非常容易，但是在这种情况下，我不确定应该从哪里开始

我尝试使用“for循环”，但我认为这并不是解决此类问题的明确答案，应该有一种非常简单的方法来解决此类问题

如果您有任何建议，我将不胜感激。提前感谢！=]

Answer 1

下面是您希望执行的代码

我包括了该函数的两个版本，一个是为了清晰，另一个是更小但功能相同

####################
# Create a list of lists / elements
####################
#x = [[1,2,3],[3,4],[3,4,[3,4,5],[4],[5,4,6]],[4,3,4,5,[[[6]]]]]
x = [[], [[[[], []], [[]], []]], [[]]]

############################################################
# First implementation (included for clarity)
############################################################
def get_len_lists(this_list):
    # Set number of elements to 0
    num_elem = 0                              

    # Loop through each element in the list...
    for elem in this_list:                          

        # .. if it's a list...
        if type(elem) == list:                      

            # ... if the list is empty, count that as an element
            if elem == list():                      
                # ... so add one
                num_elem += 1                       
            else:
                # ... get the number of elements...
                num_elem += get_len_lists(elem) + 1 

        # ... otherwise...
        else:                                       
            # ... just add one to the length of the list
            num_elem += 1                           

    # Return the number of elements in the list
    return num_elem                                 


############################################################
# Smaller implementation
############################################################
def get_len_lists_2(this_list):
    # Set number of elements to 0
    num_elem = 0                                    

    # Loop through each element in the list...
    for elem in this_list:                          

        # We add one for each level, regardless of whether it is an element or a list
        num_elem += 1                              

        # If it's a list...
        if type(elem) == list:                      

            # .. get the number of elements in the list
            num_elem += get_len_lists(elem)         

    # Return the number of elements in the list
    return num_elem                                 


result1 = get_len_lists(x) + 1
result2 = get_len_lists_2(x) + 1

print(result1)
print(result2)

输出为：

12
12

Answer 2

这可能会将您推向正确的方向：

def lenall(lst):
    if isinstance(lst, list):
        return 1 + sum(map(lenall, lst))
    return 0

>>> lenall(listlist)
12

这实际上统计数据结构中的列表总数。如果要添加其所有长度，则必须将其更改为：

def lenall(lst):
    if isinstance(lst, list):
        return len(lst) + sum(map(lenall, lst))
    return 0

>>> lenall(listlist)
11

Answer 3

下面是一个可能的解决方案：

listlist = [[], [[[[], []], [[]], []]], [[]]]

def length(lst):
    return len(lst) + sum(length(l) for l in lst if isinstance(l, list))

print(length(listlist))

输出

11