如何使用二维索引数组从三维 NumPy 数组中高效提取二维切片_网络技术

如何使用二维索引数组从三维 NumPy 数组中高效提取二维切片

发布时间 - 2026-01-30 00:00:00 点击率：次

本文介绍如何利用 `np.choose` 从形状为 (10, 33, 66) 的 3d 数组中，依据形状为 (33, 66) 的二维索引数组，沿第一维（axis=0）精确选取对应元素，最终得到形状为 (33, 66) 的结果数组。

在 NumPy 中，当需要根据每个位置的索引从高维数组中“逐点选取”元素时（即所谓的“高级索引”或“花式索引”），np.choose 是一个简洁而高效的解决方案——尤其适用于沿首个轴（axis=0）进行索引的情形。

假设我们有如下数据：

import numpy as np

# 构造示例 3D 数组：(10, 33, 66)
data_3d = np.random.random((10, 33, 66))

# 构造 2D 索引数组：每个 (i, j) 位置指定应取 data_3d[k, i, j] 中的 k 值
index_2d = np.random.randint(0, 10, size=(33, 66))  # 值域 [0, 9]

目标是生成一个形状为 (33, 66) 的二维数组 result，其中：

result[i, j] == data_3d[index_2d[i, j], i, j]

✅ 正确做法：使用 np.choose

result = np.choose(index_2d, data_3d)

⚠️ 注意事项：

np.choose 要求 index_2d 的值必须是 非负整数，且严格小于 data_3d.shape[0]（即 0 ≤ index_2d[i,j]
data_3d 会被自动视为一个包含 10 个 (33, 66) 子数组的序列（即 data_3d[0], data_3d[1], ..., data_3d[9]），np.choose 按 index_2d[i,j] 从中选择第 k 个子数组的 (i,j) 元素。
该操作等价于但显著快于显式循环或 np.stack(...)[index_2d, ...] 的广播写法（后者需构造大中间数组）。

? 替代方案（兼容性更强，推荐用于复杂场景）：若索引维度不固定或需更高灵活性（如索引其他轴），可改用 np.ogrid + 高级索引：

i_grid, j_grid = np.ogrid[:33, :66]
result = data_3d[index_2d, i_grid, j_grid]  # 形状自动广播为 (33, 66)

此写法更直观、可扩展性强，且对索引范围检查更明确（支持布尔索引、负索引等），是现代 NumPy 实践中的首选。

✅ 总结：

对「按二维索引沿第一维选取」任务，np.choose(index_2d, data_3d) 最简明；
对需调试、扩展或索引非首轴的场景，优先使用 data_3d[index_2d, i_grid, j_grid]；
始终确保索引数组 dtype 为 in
t，值域合法，避免运行时错误。

# numpy # Array # int # 循环 # 切片 # 值域 # 组中 # 是一个 # 适用于 # 更高 # 布尔 # 更强 # 首个 # 即所谓 # 快于

相关栏目：【网站优化151355 】【网络推广146373 】【网络技术251813 】【 AI营销90571 】

上一篇：2026年沪深港通春节休市时间表 2026南北向资金春节暂停

下一篇：C++ 怎么判断完全平方数 C++ sqrt取整判断逻辑【算

相关栏目网站优化
网络推广
网络技术
AI营销

最新文章 Sublime怎么一键压缩JS代码 Su sublime如何在搜索中使用正则表达式 Sublime如何设置透明窗口效果 Su mysql如何设计商品表结构_mysql css属性背景图不显示怎么办_通过检查路如何使用Golang实现排序_Golan 农历闰月是怎么回事_为合回归年加一月调整塑造《刺客信条》艾吉奥传奇的编剧离开育碧 1英里等于多少公里 1mile和km的换 css grid布局中行和列是如何定义的 PS批量旋转和翻转图片，快速校正图片方向 C# Swagger UI自定义方法 C OPPO手机九宫格和全键盘怎么切换_OP Go语言如何实现用户登录注册_Golan 1节飞行速度多少公里每小时 1节是多少公纸嫁衣8千子树第五章庙门怎么开启庙门 Laravel 多行数据编辑表单中实现逐明日之后如何提升钓鱼等级明日之后提升钓支付宝怎样查年度账单_支付宝年度账单查看 C# 多线程UI更新Dispatcher

上一篇：2026年沪深港通春节休市时间表 2026南北向资金春节暂停

下一篇：C++ 怎么判断完全平方数 C++ sqrt取整判断逻辑【算