0

0

Python嵌套列表怎么展平_多维数组降维与递归展开提取技巧

P粉602998670

P粉602998670

发布时间:2026-03-14 11:19:04

|

302人浏览过

|

来源于php中文网

原创

itertools.chain(*lst)可快速展平两层嵌套列表,但需确保子项均为可迭代对象;更安全的替代是chain.from_iterable(lst);任意深度嵌套应使用递归生成器,兼顾内存与类型安全。

python嵌套列表怎么展平_多维数组降维与递归展开提取技巧

itertools.chain 快速展平两层嵌套列表

两层嵌套(比如 [[1, 2], [3, 4], [5]])是最常见场景,itertools.chain 是最轻量、最直观的解法。它不递归,不深拷贝,性能好,也明确表达“我只是把子列表连起来”这个意图。

常见错误是误用 chain(*lst) 却没处理空子列表或非列表元素,导致 TypeError: 'int' object is not iterable

  • 确保所有子项都是可迭代对象;如果可能含 None 或数字,先过滤:[x for x in lst if isinstance(x, (list, tuple))]
  • 想兼容元组、集合等?直接用 chain.from_iterable(lst),它比 chain(*lst) 更安全,不依赖解包
  • 注意:它只展平一层,[[[1, 2]], [3]] 会变成 [[1, 2], 3],不是你想要的最终结果

处理任意深度嵌套时,递归函数比 json.loads 或正则更可靠

有人试过把嵌套列表转成字符串再用正则提取数字,或者用 json.loads(str(lst)) 强转——这两种都踩过坑:前者丢类型(所有数字变字符串),后者在含 tupleset、自定义对象时直接报 TypeError: Object of type tuple is not JSON serializable

写个简单递归函数反而更稳:

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

Peppertype.ai
Peppertype.ai

高质量AI内容生成软件,它通过使用机器学习来理解用户的需求。

下载
def flatten(obj):
    if isinstance(obj, (list, tuple, set)):
        for item in obj:
            yield from flatten(item)
    else:
        yield obj
  • yield from 避免中间列表,内存友好;要列表就包一层 list(flatten(nested))
  • 若需保留 str 不展开(比如 ['a', ['b', 'c']] 期望输出 ['a', 'b', 'c'] 而非 ['a', 'b', 'c'] 拆成字符),加判断 if isinstance(obj, str): yield obj
  • 别用 isinstance(obj, collections.abc.Iterable) —— 它把字符串、字节对象也都算进去,容易意外展开

numpy.ndarray.flatten().ravel() 的区别影响原数组修改

如果你操作的是数值型多维数组(比如图像像素、矩阵运算结果),别用 Python 原生方法,直接上 numpy。但要注意:.flatten() 返回副本,.ravel() 尽量返回视图(view),改它可能改到原数组。

  • 需要安全修改?用 arr.flatten() —— 它永远复制,不怕副作用
  • 追求性能且确定不改原数据?用 arr.ravel(),但得加 order='C' 显式指定行优先(默认行为,但显式更稳妥)
  • 遇到 AttributeError: 'list' object has no attribute 'flatten'?说明你还没转成 np.array,先调 np.array(nested_list)
  • 注意:np.array([[1,2],[3]]) 会因长度不一致降级为 dtype=object,此时 .flatten() 不会展开内部列表,得先统一结构或换回 Python 递归处理

展平后丢失索引位置?用生成器带坐标更实用

单纯展平常会丢掉“这个值原来在第几层、哪个位置”,比如做调试、映射错误行号、或构建稀疏结构时,你需要知道 7 来自 data[2][0][1]

这时候别拼字符串路径,用递归生成器返回 (value, path) 元组更灵活:

def flatten_with_path(obj, path=()):
    if isinstance(obj, (list, tuple)):
        for i, item in enumerate(obj):
            yield from flatten_with_path(item, path + (i,))
    else:
        yield obj, path
  • 调用 list(flatten_with_path(nested)) 得到像 [(1, (0, 0)), (2, (0, 1)), (3, (1, 0))] 这样的结果
  • 路径用元组而非列表,天然可哈希,方便后续做字典键或去重
  • 如果嵌套里有字典,且你想展开 dict.values(),得额外加分支:elif isinstance(obj, dict): yield from flatten_with_path(list(obj.values()), path)

展平本身不难,难的是你到底要什么:是快、是安全、是保结构、还是带上下文。选错方法,后面 debug 花的时间远超写那几行代码。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

WorkBuddy
WorkBuddy

腾讯云推出的AI原生桌面智能体工作台

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
json数据格式
json数据格式

JSON是一种轻量级的数据交换格式。本专题为大家带来json数据格式相关文章,帮助大家解决问题。

457

2023.08.07

json是什么
json是什么

JSON是一种轻量级的数据交换格式,具有简洁、易读、跨平台和语言的特点,JSON数据是通过键值对的方式进行组织,其中键是字符串,值可以是字符串、数值、布尔值、数组、对象或者null,在Web开发、数据交换和配置文件等方面得到广泛应用。本专题为大家提供json相关的文章、下载、课程内容,供大家免费下载体验。

549

2023.08.23

jquery怎么操作json
jquery怎么操作json

操作的方法有:1、“$.parseJSON(jsonString)”2、“$.getJSON(url, data, success)”;3、“$.each(obj, callback)”;4、“$.ajax()”。更多jquery怎么操作json的详细内容,可以访问本专题下面的文章。

337

2023.10.13

go语言处理json数据方法
go语言处理json数据方法

本专题整合了go语言中处理json数据方法,阅读专题下面的文章了解更多详细内容。

82

2025.09.10

if什么意思
if什么意思

if的意思是“如果”的条件。它是一个用于引导条件语句的关键词,用于根据特定条件的真假情况来执行不同的代码块。本专题提供if什么意思的相关文章,供大家免费阅读。

847

2023.08.22

js 字符串转数组
js 字符串转数组

js字符串转数组的方法:1、使用“split()”方法;2、使用“Array.from()”方法;3、使用for循环遍历;4、使用“Array.split()”方法。本专题为大家提供js字符串转数组的相关的文章、下载、课程内容,供大家免费下载体验。

761

2023.08.03

js截取字符串的方法
js截取字符串的方法

js截取字符串的方法有substring()方法、substr()方法、slice()方法、split()方法和slice()方法。本专题为大家提供字符串相关的文章、下载、课程内容,供大家免费下载体验。

221

2023.09.04

java基础知识汇总
java基础知识汇总

java基础知识有Java的历史和特点、Java的开发环境、Java的基本数据类型、变量和常量、运算符和表达式、控制语句、数组和字符串等等知识点。想要知道更多关于java基础知识的朋友,请阅读本专题下面的的有关文章,欢迎大家来php中文网学习。

1570

2023.10.24

TypeScript类型系统进阶与大型前端项目实践
TypeScript类型系统进阶与大型前端项目实践

本专题围绕 TypeScript 在大型前端项目中的应用展开,深入讲解类型系统设计与工程化开发方法。内容包括泛型与高级类型、类型推断机制、声明文件编写、模块化结构设计以及代码规范管理。通过真实项目案例分析,帮助开发者构建类型安全、结构清晰、易维护的前端工程体系,提高团队协作效率与代码质量。

26

2026.03.13

热门下载

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

精品课程

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

共4课时 | 22.5万人学习

Django 教程
Django 教程

共28课时 | 5万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.9万人学习

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

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