0

0

PythonNumpy数组操作高级技巧_矩阵运算优化案例【教学】

冷炫風刃

冷炫風刃

发布时间:2025-12-19 20:23:02

|

382人浏览过

|

来源于php中文网

原创

NumPy高效运算核心在于避免隐式拷贝、善用广播和底层优化:优先用@替代np.dot,批量运算选einsum,就地操作用out=参数,矩阵分解跳过冗余计算,并确保BLAS/LAPACK加速生效。

pythonnumpy数组操作高级技巧_矩阵运算优化案例【教学】

NumPy 数组的高效矩阵运算,核心不在“写得对”,而在“写得巧”——避开隐式拷贝、善用广播、优先调用底层优化函数,才能真正释放计算性能。

用 @ 代替 np.dot,明确语义且更高效

Python 3.5+ 支持中缀运算符 @,专为矩阵乘法设计。它比 np.dot 更简洁,且在某些场景(如多维数组批量乘法)下触发更优的底层路径。

  • 推荐写法:C = A @ B(A、B 为二维或满足广播规则的高维数组)
  • 避免嵌套 dot:np.dot(np.dot(A, B), C) → 改用 (A @ B) @ C 或直接 A @ B @ C,NumPy 会自动优化链式乘法顺序(虽不等价于手动动态规划,但可读性与效率兼顾)
  • 注意:@ 要求最后两轴满足矩阵乘法规则;若需向量内积,仍用 np.sum(a * b)a @ b.T(确保维度对齐)

批量矩阵运算:用 einsum 替代多重循环和堆叠

当需要对一批矩阵做相同运算(如每组 3×3 矩阵求逆、每对向量算外积),np.einsum 是最灵活且常更快的选择,它绕过中间数组分配,直译数学表达式。

  • 典型场景:有 N 个 3×3 矩阵 stacked 在 shape=(N,3,3) 的数组 M 中,求全部逆矩阵:
    invs = np.linalg.inv(M) ✔️(已高度优化)
    若需更复杂操作,如 “对每个 i,计算 M[i] @ v[i]”,用:
    result = np.einsum('nij,nj->ni', M, v)
  • 关键技巧:先写出指标表达式(如 'ijk,ikl->ijl' 表示 batch matmul),再验证维度是否匹配;启用 optimize=True(如 np.einsum('...ij,...jk->...ik', A, B, optimize=True))可自动选择最优收缩路径

避免不必要的副本:就地操作与 view 优先

大数组运算中,一次 .copy() 可能吃掉数 GB 内存并拖慢速度。多数 NumPy 函数支持 out= 参数,或可通过 view 复用内存。

PicWish
PicWish

推荐!专业的AI抠图修图,支持格式转化

下载

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

  • 就地更新:np.add(A, B, out=A)A = A + B 少一次内存分配;同理 np.multiply(A, scalar, out=A)
  • 谨慎使用 view:A.TA[:, ::-1] 返回 view(共享内存),但 A[::-1, :](负步长切片)在旧版本可能返回 copy;用 A.flags['OWNDATA'] 检查
  • 矩阵分解(如 SVD)默认返回新数组,若只需部分结果,用 compute_uv=False(如 s = np.linalg.svd(X, compute_uv=False))跳过 U/V 计算

混合精度与底层加速:合理启用 BLAS/LAPACK

NumPy 的线性代数函数(np.linalg)性能极度依赖后端 BLAS 库。默认安装常链接基础 OpenBLAS,但可进一步优化。

  • 验证是否生效:运行 np.show_config(),确认 blas_opt_infolapack_opt_info 显示 Intel MKL 或 OpenBLAS 路径
  • 小矩阵别硬刚:对于 shape
  • 混合精度提示:若数据允许,用 A.astype(np.float32) 降低内存带宽压力;但注意 np.linalg.inv 等对 float32 更敏感,必要时加 rcond=1e-5 提升稳定性

相关专题

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

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

755

2023.06.15

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

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

636

2023.07.20

python能做什么
python能做什么

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

759

2023.07.25

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

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

618

2023.07.31

python教程
python教程

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

1262

2023.08.03

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

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

547

2023.08.04

python eval
python eval

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

577

2023.08.04

scratch和python区别
scratch和python区别

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

708

2023.08.11

C++ 单元测试与代码质量保障
C++ 单元测试与代码质量保障

本专题系统讲解 C++ 在单元测试与代码质量保障方面的实战方法,包括测试驱动开发理念、Google Test/Google Mock 的使用、测试用例设计、边界条件验证、持续集成中的自动化测试流程,以及常见代码质量问题的发现与修复。通过工程化示例,帮助开发者建立 可测试、可维护、高质量的 C++ 项目体系。

3

2026.01.16

热门下载

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

精品课程

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

共4课时 | 0.9万人学习

Django 教程
Django 教程

共28课时 | 3.1万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.1万人学习

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

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