0

0

python怎么将一个嵌套列表扁平化_python嵌套列表扁平化方法

冰火之心

冰火之心

发布时间:2025-09-13 12:21:01

|

917人浏览过

|

来源于php中文网

原创

扁平化嵌套列表的核心是根据嵌套深度和数据规模选择合适方法:递归适用于任意深度但受限于调用栈;生成器结合yield from兼顾性能与内存;itertools.chain.from_iterable适合浅层嵌套且效率高;sum()方法简洁但性能差;列表推导式限于固定两层。处理混合类型时需用isinstance(item, list)排除字符串等可迭代对象,避免误拆。通用推荐为生成器方案,既高效又支持深层嵌套。

python怎么将一个嵌套列表扁平化_python嵌套列表扁平化方法

Python中要将嵌套列表扁平化,核心思路无非就是递归遍历,或者利用列表推导式结合一些内置函数,甚至借助一些第三方库来处理。这事儿看起来简单,但深究起来,选择哪种方法,还得看你的列表深度、数据规模以及对性能的要求。我个人觉得,没有一劳永逸的最佳方案,只有最适合特定场景的方案。

解决方案

说实话,扁平化这需求,看似简单,但实际场景中遇到的坑可不少。我通常会根据列表的嵌套深度和数据量来选择不同的策略。

递归函数:应对任意深度嵌套

这是最直观也最通用的方法,尤其当你面对一个深度不确定的嵌套列表时,递归几乎是首选。它的逻辑很简单:如果当前元素是列表,就继续“钻”进去;如果不是,就把它收集起来。

def flatten_recursive(nested_list):
    flat_list = []
    for item in nested_list:
        if isinstance(item, list):
            # 注意这里,我个人会加上一个对字符串的特殊处理,避免把字符串也当成可迭代对象拆开
            # 比如 'hello' 不应该变成 ['h', 'e', 'l', 'l', 'o']
            if isinstance(item, str): # 如果是字符串,直接添加
                flat_list.append(item)
            else: # 否则,递归处理
                flat_list.extend(flatten_recursive(item))
        else:
            flat_list.append(item)
    return flat_list

# 示例
my_nested_list = [1, [2, 3], [4, [5, 6, [7, 'hello']]], 8, 'world']
print(f"递归扁平化结果: {flatten_recursive(my_nested_list)}")
# 输出: 递归扁平化结果: [1, 2, 3, 4, 5, 6, 7, 'hello', 8, 'world']

这个方法的优点是清晰、通用,能处理任意深度的嵌套。缺点嘛,就是对于非常深的列表,可能会遇到Python的递归深度限制(默认1000层),而且函数调用的开销也相对大一些。

立即学习Python免费学习笔记(深入)”;

itertools.chain.from_iterable
:优雅且高效

itertools
模块简直是Python处理迭代器的宝藏,
chain.from_iterable
就是其中一个。它能把多个可迭代对象“链”在一起,生成一个单一的迭代器。对于只有一层嵌套的列表,它非常简洁高效。

import itertools

# 仅适用于浅层嵌套,即列表的元素本身是列表,但不再有更深层次的嵌套
shallow_nested_list = [[1, 2], [3, 4], [5]]
flat_list_itertools = list(itertools.chain.from_iterable(shallow_nested_list))
print(f"itertools.chain扁平化结果 (浅层): {flat_list_itertools}")
# 输出: itertools.chain扁平化结果 (浅层): [1, 2, 3, 4, 5]

但如果你的列表是深层嵌套的,比如

[1, [2, [3, 4]]]
,直接用
itertools.chain.from_iterable
是不行的。这时候,你可以结合递归思想,用一个生成器函数来处理:

import itertools

def flatten_generator(nested_list):
    for item in nested_list:
        if isinstance(item, list) and not isinstance(item, (str, bytes)): # 排除字符串被迭代
            yield from flatten_generator(item) # 使用 yield from 简化递归生成器
        else:
            yield item

# 示例
my_nested_list = [1, [2, 3], [4, [5, 6, [7, 'hello']]], 8, 'world']
flat_list_gen = list(flatten_generator(my_nested_list))
print(f"生成器扁平化结果: {flat_list_gen}")
# 输出: 生成器扁平化结果: [1, 2, 3, 4, 5, 6, 7, 'hello', 8, 'world']

这个生成器方法非常棒,它结合了递归的通用性和生成器的内存效率。

yield from
语法更是让代码简洁不少。它在需要时才生成元素,不会一次性在内存中构建整个扁平列表,对大数据量非常友好。

sum()
函数:一个“奇技淫巧”(慎用)

你可能听说过用

sum([], list_of_lists)
来扁平化列表。这个方法确实很简洁,但它有一些限制和潜在的问题。

# 仅适用于浅层嵌套
shallow_nested_list = [[1, 2], [3, 4], [5]]
flat_list_sum = sum(shallow_nested_list, [])
print(f"sum() 扁平化结果 (浅层): {flat_list_sum}")
# 输出: sum() 扁平化结果 (浅层): [1, 2, 3, 4, 5]

这里的

[]
sum()
函数的起始值,每次迭代都会把子列表加到这个空列表中。它的问题在于:

  1. 性能:每次加法操作都会创建一个新的列表对象,对于大型列表,这会非常低效。
  2. 深度限制:和
    itertools.chain.from_iterable
    直接使用一样,它也只能处理一层嵌套。
  3. 可读性:对于不熟悉这种用法的人来说,
    sum()
    用来扁平化列表有点“反直觉”,可读性不如其他方法。

所以,我个人建议,除非你明确知道列表很小且只有一层嵌套,并且追求极致的简洁(或者说,是炫技),否则不推荐使用

sum()

列表推导式(特定场景)

如果你的嵌套深度是固定的,并且你知道具体是几层,列表推导式也能派上用场。

# 两层嵌套
two_level_nested = [[1, 2], [3, 4, [5, 6]]] # 哎呀,这里依然是深层,列表推导式直接搞不定
# 应该是这样:
two_level_nested_fixed = [[1, 2], [3, 4], [5, 6]]
flat_list_comp = [item for sublist in two_level_nested_fixed for item in sublist]
print(f"列表推导式扁平化结果 (固定两层): {flat_list_comp}")
# 输出: 列表推导式扁平化结果 (固定两层): [1, 2, 3, 4, 5, 6]

对于固定两层的情况,这很清晰。但如果嵌套深度不固定,或者超过两层,列表推导式就会变得非常复杂,甚至无法实现。所以,它的应用场景比较有限。

FineVoice
FineVoice

FineVoice是一种AI数字语音解决方案,可以帮助用户增强声音,并配有实时变声器

下载

扁平化嵌套列表时,性能和内存效率该如何权衡?

这确实是个值得深思的问题。在我看来,选择哪种扁平化方法,很大程度上取决于你对性能和内存的需求。

  • sum([], list_of_lists)
    :这个方法在性能上通常是最差的,因为它在每次迭代中都会创建新的列表对象。如果你的列表非常大,或者需要频繁扁平化,它会成为一个性能瓶颈。内存方面,因为它不断创建新列表,也可能导致较高的瞬时内存占用。所以,除非数据量极小,我基本不会考虑它。

  • 递归函数 (直接返回列表):通用性最好,能处理任意深度。但它的性能开销主要来自函数调用栈的创建和销毁。对于非常深的嵌套,可能会触发Python的递归深度限制(

    RecursionError
    ),你需要手动修改这个限制,但这本身就是一种风险。内存方面,它会一次性构建并返回一个完整的扁平列表,如果原始列表非常大,这个最终的扁平列表也可能占用大量内存。

  • itertools.chain.from_iterable
    (配合生成器):这是我个人最推荐的组合。
    itertools
    模块是用C语言实现的,因此它的效率极高。当结合生成器函数 (
    yield from
    ) 使用时,它能实现惰性求值。这意味着它不会一次性把所有扁平化后的元素都加载到内存中,而是在你每次需要一个元素时才生成它。这对于处理海量数据或非常深的嵌套列表来说,是内存效率最高的方案。你可以在处理过程中逐步消费这些元素,而无需等待整个列表扁平化完成。性能上,由于C语言实现和迭代器特性,它通常比纯Python的递归函数更快。

  • 列表推导式 (固定深度):如果深度固定且很浅(比如两层),列表推导式通常性能不错,因为它也是高度优化的。内存方面,它会一次性构建整个扁平列表。

我的个人权衡是: 如果列表不大,嵌套深度也不确定,递归函数(直接返回列表)最简单直接。 如果列表很大,或者内存是关键因素,那么结合

itertools.chain.from_iterable
的生成器方案是王道。 如果列表只有一层嵌套,且数据量适中,
itertools.chain.from_iterable
直接用
list()
包裹,简洁高效。 总之,多数情况下,生成器是兼顾性能和内存的最佳选择。

面对不同深度的嵌套列表,有哪些通用的扁平化策略?

对于不同深度的嵌套列表,我们追求的“通用策略”核心在于能够处理任意未知深度

  1. 递归生成器函数(最通用且推荐) 这是我个人认为最优雅、最通用的方案。无论是浅层还是深层嵌套,只要是合法的列表结构,它都能应对。

    def generic_flatten(iterable):
        for element in iterable:
            # 这里的判断很重要,要确保只对真正的列表进行递归,
            # 避免将字符串、元组等其他可迭代对象也拆开
            if isinstance(element, list) and not isinstance(element, (str, bytes)):
                yield from generic_flatten(element)
            else:
                yield element
    
    # 示例:
    list_a = [1, [2, 3], 4]
    list_b = [1, [2, [3, [4, 5]]], 6, 'hello', [7, [8]]]
    list_c = [1, (2, 3), [4, {5:6}], 7] # 包含其他可迭代对象
    
    print(f"通用扁平化 list_a: {list(generic_flatten(list_a))}")
    print(f"通用扁平化 list_b: {list(generic_flatten(list_b))}")
    # 注意,generic_flatten 默认只扁平化列表,元组、字典等不会被拆开
    print(f"通用扁平化 list_c: {list(generic_flatten(list_c))}")

    这个策略的强大之处在于它的灵活性和惰性求值,无论列表有多深、多复杂,它都能按需生成扁平化的元素。

  2. 基于栈的迭代方法(避免递归深度限制) 当递归深度非常深,担心Python的递归限制时,可以考虑使用一个显式的栈来模拟递归过程。这本质上是把递归调用转换为迭代循环。

    def iterative_flatten(nested_list):
        result = []
        stack = list(nested_list) # 将顶层列表的元素放入栈中
        while stack:
            item = stack.pop(0) # 从栈顶取出一个元素
            # 同样,只对列表进行处理,排除字符串
            if isinstance(item, list) and not isinstance(item, (str, bytes)):
                # 如果是列表,将其元素逆序放回栈中,保证取出顺序
                stack = list(item) + stack # 注意这里,为了保持原有顺序,需要将新元素放在前面
            else:
                result.append(item)
        return result
    
    # 示例
    deep_nested_list = [1, [2, [3, [4, [5, [6, [7, [8, [9, [10]]]]]]]]]]
    print(f"迭代扁平化结果: {iterative_flatten(deep_nested_list)}")

    这个迭代方法避免了递归深度限制,但代码实现上会稍微复杂一些,尤其是在处理元素顺序时需要特别注意栈的操作。我这里为了保持原有顺序,用了

    stack = list(item) + stack
    这种方式,它在性能上可能不如直接
    extend
    。如果对顺序不敏感,或者可以反转处理,会有更高效的栈操作方式。但通常情况下,生成器方案已经足够应对大多数需求了。

扁平化过程中,如何处理非列表类型元素,或者混合数据类型?

这是一个非常实际的问题,因为真实的列表往往不会那么“纯粹”,里面可能混杂着数字、字符串、元组、字典甚至自定义对象。我的通用策略是:只扁平化列表类型,对其他类型保持原样。

  1. 精确判断类型:

    isinstance(item, list)
    在我的
    flatten_recursive
    flatten_generator
    函数中,关键在于
    isinstance(item, list)
    这个判断。它确保只有当一个元素确实是
    list
    类型时,我们才尝试对其进行递归处理。

  2. 排除字符串被错误扁平化:

    and not isinstance(item, (str, bytes))
    这是我特别强调的一点。Python中的字符串
    str
    也是可迭代对象。如果你仅仅判断
    isinstance(item, Iterable)
    而不排除
    str
    ,那么
    'hello'
    就会被扁平化成
    ['h', 'e', 'l', 'l', 'o']
    ,这通常不是我们想要的结果。所以,在判断一个元素是否需要递归时,务必加上对
    str
    (以及
    bytes
    ) 类型的排除。

    # 再次强调这个关键的判断逻辑
    if isinstance(element, list) and not isinstance(element, (str, bytes)):
        # 只有满足这两个条件,才进行递归扁平化
        yield from generic_flatten(element)
    else:
        # 否则,直接将其作为扁平化列表的一部分
        yield element
  3. 处理其他混合类型(元组、字典、集合等) 如果你的列表里包含了元组、字典、集合等其他可迭代对象,而你不希望它们被扁平化(比如你希望

    (1, 2)
    保持为一个元组,而不是
    1, 2
    两个独立的元素),那么上述的
    isinstance(element, list)
    策略就是正确的。它会把这些非列表的可迭代对象视为单个元素直接添加。

    mixed_list = [1, [2, 3], (4, 5), {'a': 6}, 7, "text"]
    # 使用 generic_flatten
    flat_result = list(generic_flatten(mixed_list))
    print(f"混合类型扁平化结果: {flat_result}")
    # 期望输出: [1, 2, 3, (4, 5), {'a': 6}, 7, 'text']
    # 实际输出也是如此,因为 (4,5) 和 {'a':6} 不是 list 类型,不会被进一步拆解
  4. 如果需要扁平化所有可迭代对象(除了字符串) 在某些特殊场景下,你可能希望将所有可迭代对象(除了字符串)都扁平化。这时,你可以修改判断条件:

    from collections.abc import Iterable
    
    def flatten_all_iterables(nested_list):
        for item in nested_list:
            # 判断是否是可迭代对象,但排除字符串和字节串
            if isinstance(item, Iterable) and not isinstance(item, (str, bytes)):
                yield from flatten_all_iterables(item)
            else:
                yield item
    
    # 示例
    mixed_list_with_tuple = [1, [2, 3], (4, 5), {'a': 6}, 7, "text"]
    print(f"扁平化所有可迭代对象结果: {list(flatten_all_iterables(mixed_list_with_tuple))}")
    # 期望输出: [1, 2, 3, 4, 5, 'a', 6, 7, 'text']
    # 字典 {'a': 6} 会被扁平化为 'a', 6 (因为字典迭代时先出键,再出值)
    # 这可能不是你想要的,所以要慎重!

    这个例子说明了,如果你不加区分地扁平化所有

    Iterable
    ,可能会得到意想不到的结果,特别是对于字典,它会迭代出键。所以,多数情况下,只针对
    list
    类型进行扁平化
    是最安全、最符合预期的策略。

总结来说,处理混合数据类型时,关键在于你对“扁平化”的定义:是只针对列表结构,还是针对所有可迭代对象?明确这一点,然后用

isinstance
进行精确控制,就能写出健壮的代码。

相关专题

更多
python开发工具
python开发工具

php中文网为大家提供各种python开发工具,好的开发工具,可帮助开发者攻克编程学习中的基础障碍,理解每一行源代码在程序执行时在计算机中的过程。php中文网还为大家带来python相关课程以及相关文章等内容,供大家免费下载使用。

765

2023.06.15

python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

639

2023.07.20

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

764

2023.07.25

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

619

2023.07.31

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

1285

2023.08.03

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

549

2023.08.04

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

579

2023.08.04

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

709

2023.08.11

PS使用蒙版相关教程
PS使用蒙版相关教程

本专题整合了ps使用蒙版相关教程,阅读专题下面的文章了解更多详细内容。

52

2026.01.19

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 6.3万人学习

Django 教程
Django 教程

共28课时 | 3.3万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.2万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号