计算嵌套列表的深度或最深层次

Question

一个作业有一个真正的问题（和头痛）…

我在入门编程班上，我必须写一个函数，给定一个列表，该函数将返回其达到的“最大”深度。例如：[1,2,3]将返回1，[ 1，[2,3]]将返回2 …

我已经编写了这段代码（这是我能得到的最好的T_T）

def flat(l):
    count=0
    for item in l:
        if isinstance(item,list):
            count+= flat(item)
    return count+1

但是，它显然不像应有的那样工作，因为如果存在不计入最大深度的列表，它仍然会增加计数器…

例如：当我将函数与[1,2，[3,4]，5，[6]，7]一起使用时，它应该返回2，但它返回3 …

任何想法或帮助将不胜感激^^非常感谢！我已经为此困扰了好几个星期…

Answer 1

广度优先，无需递归，它也适用于其他序列类型：

from collections import Sequence
from itertools import chain, count

def depth(seq):
    for level in count():
        if not seq:
            return level
        seq = list(chain.from_iterable(s for s in seq if isinstance(s, Sequence)))

相同的想法，但内存消耗少得多：

from collections import Sequence
from itertools import chain, count

def depth(seq):
    seq = iter(seq)
    try:
        for level in count():
            seq = chain([next(seq)], seq)
            seq = chain.from_iterable(s for s in seq if isinstance(s, Sequence))
    except StopIteration:
        return level