首页 >后端开发 >Python教程 >我们如何在 Python 中有效地展平任意嵌套列表?

我们如何在 Python 中有效地展平任意嵌套列表?

DDD
DDD原创
2024-12-28 16:04:11363浏览

How Can We Efficiently Flatten Arbitrarily Nested Lists in Python?

任意嵌套列表展平

具有不同深度级别的嵌套列表在将其减少到单一维度方面提出了挑战。虽然有许多解决方案可以展平浅列表,但许多解决方案都在处理不规则的嵌套列表,例如 [[[1, 2, 3], [4, 5]], 6]。

递归方法

问题中提到的一种方法涉及递归:

def flatten(x):
    result = []
    for el in x:
        if hasattr(el, "__iter__") and not isinstance(el, basestring):
            result.extend(flatten(el))
        else:
            result.append(el)
    return result

此算法迭代列表,递归地展平任何嵌套列表并将不可迭代的元素附加到结果中。

用于提高可读性和性能的生成器函数

生成器函数提供了一种替代方法,可以增强代码的可读性和效率我们的扁平化过程。

Python 2

from collections import Iterable

def flatten(xs):
    for x in xs:
        if isinstance(x, Iterable) and not isinstance(x, basestring):
            for item in flatten(x):
                yield item
        else:
            yield x

Python 3

from collections.abc import Iterable

def flatten(xs):
    for x in xs:
        if isinstance(x, Iterable) and not isinstance(x, (str, bytes)):
            yield from flatten(x)
        else:
            yield x

在 Python 3 中,yield from 运算符可以方便地按顺序从嵌套生成器返回项目,而在 Python 2 中,我们显式地迭代子生成器。

以上是我们如何在 Python 中有效地展平任意嵌套列表?的详细内容。更多信息请关注PHP中文网其他相关文章!

声明:
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn