0

0

Python多维数组怎么压平_递归展开与yield from生成器

P粉602998670

P粉602998670

发布时间:2026-03-17 01:59:13

|

201人浏览过

|

来源于php中文网

原创

numpy.ndarray.flatten()最快但返回副本;需递归扁平化嵌套list时,应显式类型判断+itertools.chain.from_iterable(),避免字符串误拆;混用numpy与list需先统一类型再处理。

python多维数组怎么压平_递归展开与yield from生成器

numpy.ndarray.flatten() 最快,但要注意返回副本

多数时候你不是真要“递归展开”,只是想把多维 np.array 变成一维。直接调 .flatten() 就行,它底层走 C 实现,比任何 Python 循环都快。

但它默认返回新数组,不修改原数组;如果传 order='F'(列优先),结果和 .ravel() 不同,容易混淆形状顺序。

  • a = np.array([[1, 2], [3, 4]])a.flatten() 返回 array([1, 2, 3, 4])
  • a.flatten('F') 返回 array([1, 3, 2, 4]),对应 Fortran 排序
  • 若想复用内存、允许返回视图,改用 .ravel(),但遇到不可连续内存时会悄悄拷贝

嵌套 list 怎么递归压平?别手写递归,用 itertools.chain.from_iterable() 配合 isinstance()

Python 原生 list 嵌套层级不固定时,yield from 看着优雅,但容易栈溢出或漏判字符串——因为字符串也是可迭代对象,会被误拆成字符。

更稳的写法是显式判断类型,配合 itertools.chain.from_iterable() 扁平化一层,再循环处理,避免递归深度问题。

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

  • 错误示范:def flatten(x): yield from (flatten(i) if hasattr(i, '__iter__') else [i] for i in x) → 字符串、字节串全中招
  • 正确判断:isinstance(x, (list, tuple, set, frozenset)),排除 strbytes
  • 简单场景可用 sum(nested_list, []),但只适用于两层且元素全是 list,性能差、易报错

yield from 真要用?必须加类型守卫 + 深度限制

如果你坚持用生成器递归(比如处理超大嵌套结构、不想一次性加载到内存),yield from 可以,但必须设防。

Flowith
Flowith

一款GPT4驱动的节点式 AI 创作工具

下载

核心问题是:Python 默认递归限制约 1000 层,深层嵌套直接 RecursionError;同时得防止把 dict、range、自定义迭代器等非容器类型误展开。

  • 加守卫:if isinstance(item, (list, tuple)) and not isinstance(item, (str, bytes)):
  • 加深度参数:def flatten(x, depth=0, max_depth=10),到顶就 return
  • 别对 dict 直接 yield from d——那只会 yield key,要展开值得用 d.values()

混用 numpy 和 list 嵌套?先统一类型再处理

常见坑:函数接收的是 [np.array([1,2]), [3,4]] 这种混合结构,既不能直接 .flatten(),又不能用纯 Python 递归安全处理——因为 np.ndarray 的布尔值判断会报 ValueError: The truth value of an array with more than one element is ambiguous

解决思路不是硬刚类型判断,而是提前归一化:把所有子项转成 list 或统一转成 np.ndarray,再选对应方法。

  • 转 list:[x.tolist() if hasattr(x, 'tolist') else x for x in mixed],再进递归扁平逻辑
  • 转 ndarray:np.concatenate([np.atleast_1d(x) for x in mixed]),但要求所有子项能广播为一维
  • 最省事:用 np.hstack([x.ravel() if hasattr(x, 'ravel') else np.array(x).ravel() for x in mixed])

混类型结构没有银弹,关键在压平前明确数据契约——是用户传进来就该保证同构,还是你的函数必须兜底。后者代价高,容易藏 bug。

热门AI工具

更多
DeepSeek
DeepSeek

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

豆包大模型
豆包大模型

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

WorkBuddy
WorkBuddy

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

腾讯元宝
腾讯元宝

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

文心一言
文心一言

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

讯飞写作
讯飞写作

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

即梦AI
即梦AI

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

ChatGPT
ChatGPT

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

相关专题

更多
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

字符串介绍
字符串介绍

字符串是一种数据类型,它可以是任何文本,包括字母、数字、符号等。字符串可以由不同的字符组成,例如空格、标点符号、数字等。在编程中,字符串通常用引号括起来,如单引号、双引号或反引号。想了解更多字符串的相关内容,可以阅读本专题下面的文章。

651

2023.11.24

java读取文件转成字符串的方法
java读取文件转成字符串的方法

Java8引入了新的文件I/O API,使用java.nio.file.Files类读取文件内容更加方便。对于较旧版本的Java,可以使用java.io.FileReader和java.io.BufferedReader来读取文件。在这些方法中,你需要将文件路径替换为你的实际文件路径,并且可能需要处理可能的IOException异常。想了解更多java的相关内容,可以阅读本专题下面的文章。

1249

2024.03.22

php中定义字符串的方式
php中定义字符串的方式

php中定义字符串的方式:单引号;双引号;heredoc语法等等。想了解更多字符串的相关内容,可以阅读本专题下面的文章。

1206

2024.04.29

go语言字符串相关教程
go语言字符串相关教程

本专题整合了go语言字符串相关教程,阅读专题下面的文章了解更多详细内容。

194

2025.07.29

chatgpt使用指南
chatgpt使用指南

本专题整合了chatgpt使用教程、新手使用说明等等相关内容,阅读专题下面的文章了解更多详细内容。

0

2026.03.16

热门下载

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

精品课程

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

共4课时 | 22.5万人学习

Django 教程
Django 教程

共28课时 | 5.1万人学习

SciPy 教程
SciPy 教程

共10课时 | 2万人学习

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

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