计算嵌套列表的深度或最深级别

Question

A 在作业上遇到了真正的问题（并且令人头疼）...

我正在上入门编程课程，我必须编写一个函数，给定一个列表，该函数将返回它所达到的“最大”深度.. 。 例如： [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

这是编写函数的一种方法

depth = lambda L: isinstance(L, list) and max(map(depth, L))+1

，我认为您缺少的想法是使用 max()

Answer 2

让我们首先稍微重新表述一下您的要求。

列表的深度比其子列表的最大深度大一。

现在，这可以直接转换为代码：

def depth(l):
    if isinstance(l, list):
        return 1 + max(depth(item) for item in l)
    else:
        return 0

Answer 3

简单的递归

def flat(l):
    depths = []
    for item in l:
        if isinstance(item, list):
            depths.append(flat(item))
    if len(depths) > 0:
        return 1 + max(depths)
    return 1

Answer 4

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

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

Answer 5

用Python的一行代码完成的:)

享受

def f(g,count=0): return count if not isinstance(g,list) else max([f(x,count+1) for x in g])

Answer 6

一种不需要任何额外模块并且无论深度如何都具有相同速度的方法：

def depth(nested):
    instring = False
    count = 0
    depthlist = []
    for char in repr(nested):
        if char == '"' or char == "'":
            instring = not instring
        elif not instring and ( char == "[" or char == ")" ):
            count += 1
        elif not instring and ( char == "]" or char == ")" ):
            count -= 1
        depthlist.append(count)
    return(max(depthlist))

基本上，它的作用是使用 repr() 将列表转换为字符串。然后，对于该字符串中等于“(”或“[”的每个字符，它会增加变量count。对于右括号，它会减少<然后返回 count 已达到的最大值。

Answer 7

辱骂方式：
假设您的列表名为 mylist

mybrackets = map(lambda x: 1 if x=='[' else -1, [x for x in str(mylist) if x=='[' or x==']'])  
maxdepth = max([sum(mybrackets[:i+1]) for i in range(len(mybrackets))])

这会将您的列表转换为左括号和右括号列表，然后查找在相应右括号出现之前出现的最大数量的左括号。

Answer 8

让我们坚持使用原始的、相当优雅的解决方案，并使用 max。许多其他解决方案都使用 lambda 表达式或需要外部库。这个使用“纯”Python 并保持简单。

def depth(lst):
    d = 0
    for item in lst:
        if isinstance(item, list):
            d = max(depth(item), d)
    return d + 1

奖励：此解决方案对空列表进行计数，并避免了遍历字符串的陷阱（假设您想对空列表进行计数）。

Answer 9

在Numpy中，您可以将数据结构转换为numpy数组并使用其库函数。 arr.shape 给出每层的长度，因此我们可以 len() 形状并获取结构的深度：

import numpy as np

def f( lists ):
  arr = np.array( lists )
  return len(arr.shape)

f( [[[1,2],[3,4]],[[3,4],[5,6]]] ) # results in 3
f( [[1,2],[3,4]] ) # results in 2
f( [1,2] )  # results in 1
f( [] )  # results in 1
# -- Be dragons: --
f( [[1],[3,4]] ) # results in 1

形状的 Numpy 文档：https://docs.scipy.org/doc/numpy/reference/ generated/numpy .ndarray.shape.html

Answer 10

我为每个可迭代对象扩展了 hammar 的答案（默认情况下禁用字符串）：

def depth(arg, exclude=None):
    if exclude is None:
        exclude = (str, )

    if isinstance(arg, tuple(exclude)):
        return 0

    try:
        if next(iter(arg)) is arg:  # avoid infinite loops
            return 1
    except TypeError:
        return 0

    try:
        depths_in = map(lambda x: depth(x, exclude), arg.values())
    except AttributeError:
        try:
            depths_in = map(lambda x: depth(x, exclude), arg)
        except TypeError:
            return 0

    try:
        depth_in = max(depths_in)
    except ValueError:
        depth_in = 0

    return 1 + depth_in

Answer 11

如果您正在寻找快速修复方法

def getDepth(matrix):
    try:
        len(matrix)
        return getDepth(matrix[0]) + 1
    except:
        return 0

Answer 12

对上述内容进行简短补充，以便它也可以处理空列表：

def list_depth(list_of_lists):
    if isinstance(list_of_lists, list):
        if(len(list_of_lists) == 0):
            depth = 1
        else:
            depth = 1 + max([list_depth(l) for l in list_of_lists])
    else:
        depth = 0
    return depth

Answer 13

您也可以仅使用 python 递归地执行此操作：

def depth(L,d):
    max = d
    for i in range(len(L)):
        if type(L[i])==list :
            a = depth(L[i],d+1)
            if a>max :
                 max = a
    return(max)

这个函数是递归的，它的作用是它只是进入列表的最大深度，计算列表的深度，当它爬回来时，它只保留最大的深度在所有嵌套列表中注册。

Answer 14

@John 的解决方案非常好，但是要解决空列表情况，例如 []、[[]] 或进一步嵌套的列表，您可能需要执行类似的操作

depth = lambda L: isinstance(L, list) and (max(map(depth, L)) + 1) if str(L) else 1