使用NumPy矢量化高效查找分割子数组的最大值

---

本文旨在探讨如何利用NumPy的矢量化能力，高效地从一个一维数组中查找由指定索引分割而成的各个子数组的最大值，避免传统的Python循环和显式数组分割操作。核心解决方案是运用`np.maximum.reduceat`函数，并强调了在索引数组中包含起始索引0的重要性，以确保所有子数组都能被正确处理，从而显著提升性能和代码简洁性。

在数据处理和科学计算中，我们经常需要对数组的特定分段进行聚合操作，例如查找最大值、最小值或求和。当这些分段由一系列分割点定义时，一个常见的需求是获取每个分段的聚合结果。虽然Python的for循环结合numpy.split可以实现这一目标，但对于大型数组，这种方法效率较低，因为它涉及多次数组创建和Python层面的迭代。

问题描述

假设我们有一个一维NumPy数组arr和一组分割索引ind。我们希望找出通过这些索引分割后形成的每个子数组的最大值。

例如：

import numpy as np

arr = np.arange(12)
ind = np.array([3, 5, 9])

# 使用传统方法：先分割，再循环
sub_arrays = np.split(arr, ind)
# 结果: [array([0, 1, 2]), array([3, 4]), array([5, 6, 7, 8]), array([ 9, 10, 11])]

max_values_loop = [sub_array.max() for sub_array in sub_arrays]
# 结果: [2, 4, 8, 11]

这种方法虽然直观，但其性能瓶颈在于np.split会创建多个新的数组对象，并且列表推导式在Python层面上执行迭代。对于追求高性能的NumPy应用，我们需要一种更“矢量化”的解决方案。

矢量化解决方案：使用 np.maximum.reduceat

NumPy提供了一个强大的函数np.ufunc.reduceat，它允许我们对数组的指定“块”或“段”执行通用函数（ufunc）的累积操作。对于查找每个子数组的最大值，我们可以利用np.maximum.reduceat。

np.maximum.reduceat(array, indices) 的工作原理是，它在array中从indices数组中的每个索引位置开始，对后续元素执行maximum操作，直到遇到下一个索引位置（或数组末尾）。每个indices中的索引都会作为新段的起始点。

关键步骤：调整索引数组

百度文心百中

百度大模型语义搜索体验中心

251 查看详情

为了让np.maximum.reduceat产生与np.split后列表推导式相同的效果，我们需要对ind数组进行一个关键的调整：必须在ind数组的开头添加索引0。

这是因为np.maximum.reduceat会从indices数组中的每个元素指定的位置开始一个新的累积操作。如果我们不包含0，那么数组的第一个分段（从索引0到ind[0]-1）将不会被正确地作为一个独立的段来处理。通过添加0，我们明确指示从数组的起始位置开始第一个分段的最大值查找。

实现示例：

import numpy as np

arr = np.arange(12)
ind = np.array([3, 5, 9])

# 1. 准备用于 reduceat 的索引数组
# 必须包含起始索引 0
indices_for_reduceat = np.concatenate(([0], ind))
# 结果: [ 0  3  5  9]

# 2. 使用 np.maximum.reduceat 计算每个段的最大值
max_values_vectorized = np.maximum.reduceat(arr, indices_for_reduceat)

print(f"原始数组: {arr}")
print(f"分割索引: {ind}")
print(f"矢量化计算的最大值: {max_values_vectorized}")
# 预期输出: [ 2  4  8 11]

# 验证与传统方法的输出一致性
# max_values_loop = [sub_array.max() for sub_array in np.split(arr, ind)]
# print(f"传统方法计算的最大值: {max_values_loop}")

运行上述代码，max_values_vectorized将输出[ 2 4 8 11]，与传统方法的结果完全一致。

np.maximum.reduceat 的工作机制详解

让我们以上述例子为例，理解np.maximum.reduceat(arr, [0, 3, 5, 9])是如何工作的：

1. 第一个段（索引 0）： 从arr[0]开始，计算arr[0]到arr[3-1]（即arr[0:3]）的最大值。结果是max(0, 1, 2) = 2。
2. 第二个段（索引 3）： 从arr[3]开始，计算arr[3]到arr[5-1]（即arr[3:5]）的最大值。结果是max(3, 4) = 4。
3. 第三个段（索引 5）： 从arr[5]开始，计算arr[5]到arr[9-1]（即arr[5:9]）的最大值。结果是max(5, 6, 7, 8) = 8。
4. 第四个段（索引 9）： 从arr[9]开始，计算arr[9]到数组末尾（即arr[9:12]）的最大值。结果是max(9, 10, 11) = 11。

最终，所有这些最大值被收集到一个新的NumPy数组中，作为np.maximum.reduceat的返回值。

优点与注意事项

• 性能提升： np.maximum.reduceat是高度优化的C语言实现，避免了Python层面的循环和中间数组的创建，对于大数据集能够提供显著的性能优势。
• 代码简洁： 将多行循环代码精简为一行NumPy函数调用，提高了代码的可读性和维护性。
• 通用性： ufunc.reduceat不仅限于maximum，还可以应用于其他NumPy通用函数，如np.add.reduceat（求和）、np.minimum.reduceat（最小值）、np.multiply.reduceat（乘积）等，使其成为处理分段聚合问题的强大工具。
• 索引的精确性： 务必确保indices数组是排序的，并且包含所有期望的起始点。特别是，如果需要从数组开头开始处理，0必须作为第一个索引包含在内。

总结

通过巧妙地使用np.maximum.reduceat并正确构造索引数组（即在原始分割索引前添加0），我们可以高效且矢量化地解决查找NumPy数组分割子段最大值的问题。这种方法不仅提升了计算性能，也使得代码更加符合NumPy的编程范式，是处理类似分段聚合任务时的推荐实践。掌握ufunc.reduceat的用法，将极大地增强您在NumPy中处理复杂数据操作的能力。

以上就是使用NumPy矢量化高效查找分割子数组的最大值的详细内容，更多请关注其它相关文章！

# 几种  # seo推广php  # 淘宝seo开店流程  # 唐山seo外贸官网  # 佛山seo网络营销推广价格  # 莞城家装网站推广招聘  # 招生网站如何进行优化  # 正规seo推广服务公司  # 机器人推广营销案例论文  # 如何让seo值增大  # 电商关键词匹配软件排名  # 迭代  # 最小值  # 这种方法  # python  # 百中  # 浮点  # 结果是  # 组中  # 矢量化  # 第一个  # red  # numpy函数  # 性能瓶颈  # 工具  # 大数据  # c语言 

相关栏目： 【 Google疑问12 】 【 Facebook疑问10 】 【 优化推广96088 】 【 技术知识133117 】 【 IDC资讯59369 】 【 网络运营7196 】 【 IT资讯61894 】

相关推荐： 掌握产品代码正则表达式：避免常见陷阱与精确匹配  原子笔记app误删找回教程  哔哩哔哩黑名单怎么查看  宝妈做视频号该写什么标签话题？宝妈关注的话题有哪些？  晨报|开发商暗示《空洞骑士：丝之歌》DLC开发中 《合金装备4》有望重制  附近酒吧怎么找？  cad加载的线型看不见怎么办_cad线型不可见问题解决方法  店铺如何做视频号推广？做视频号推广有用吗？  Mac如何开启画中画模式_Mac Safari浏览器视频画中画功能  win11怎么设置默认终端为Windows Terminal Win11替代CMD和PowerShell【技巧】  Go Template中优雅处理循环最后一项：自定义函数实践  Magento 2 产品保存事件中安全更新属性的最佳实践  QQ阅读小说搜索入口地址_QQ阅读小说搜索入口地址搜索在线阅读  响应式设计中动态背景颜色条的实现指南  《磁力猫》最好用的磁官网  顺丰快递怎么查物流_顺丰快递物流信息实时查询操作指南  J*aScript与HTML元素交互：图片点击事件与链接处理教程  C++怎么解决数值计算中的精度问题_C++浮点数误差与数值稳定性分析  Windows Audio服务启动失败怎么办_电脑没声音的终极服务修复法【修复】  使用document.execCommand实现Web文本编辑器加粗/取消加粗  京东物流快递破损了怎么办_京东快递破损理赔流程  《tt语音》超级玩家开通方法  J*aScript对象中深度嵌套URL键的查找与更新策略  抖音号已注销怎么解绑企业认证？不解绑企业认证会怎样？  Win10共享文件夹设置方法 Win10局域网文件共享全攻略【教程】  修复UI元素交互障碍：从“开始”按钮到信息框的平滑过渡实现  PHP动态导航按钮：根据用户登录状态切换链接与文本  汽水音乐官网网页版入口 汽水音乐官网网页版在线入口  《伊瑟》凶影追缉库卢鲁boss攻略  如何在CSS中使用过渡制作按钮边框渐变_border-color transition实现  cad怎么隐藏指定的图层_cad隐藏或冻结图层方法  以下哪一个是适应长期护理制度发展而设立的新职业  在Flask应用中安全高效地更新SQLAlchemy用户数据  QQ网页版官方账号登录入口 QQ网页版网页版入口快速导航  《海贝音乐》均衡器设置方法  优化响应式标题底部边框：CSS实现技巧与最佳实践  macosmonterey系统外接显示器驱动怎么安装_macosmonterey外接显示器驱动与分辨率调整  CodeIgniter 3 中基于 MySQL 数据高效生成动态图表教程  PHP安全加载非公开目录图片与动态内容类型处理指南  PHP odbc_fetch_array 返回值处理：如何正确访问嵌套数组元素  英雄联盟争者留名活动介绍  Linux如何开发轻量级数据服务模块_Linux服务化设计  Mac怎么关闭按键声音_Mac键盘打字音效设置  百度浏览器无法安装扩展程序_百度浏览器插件安装失败原因解析  Win11怎么开启HDR_Windows 11显示器画质增强设置  J*aScript桌面应用_Electron多进程架构实战  C#中的Record类型有什么优势？C# 9新特性Record与Class的用法区别  word文档行距怎么调？word文档调行距的操作步骤  《淘票票》添加到苹果钱包教程  智慧团建活动报名入口 智慧团建活动报名入口手机端官网​ 

 2025-11-30