我想使用另一个索引数组提取给定多维 numpy 数组的元素。但是它的行为并不符合我的预期。下面是一个简单的例子：
import numpy as np a = np.random.random((3, 3, 3)) idx = np.asarray([[0, 0, 0], [0, 1, 2]]) b = a[idx] print(b.shape) # expect (2, ), got (2, 3, 3, 3)
为什么会这样呢？我应该如何修改代码以使 b 仅包含两个元素：a[0, 0, 0] 和 a[0, 1, 2]？
您正在寻找 numpy 高级索引
https://www.php.cn/link/2d661a763280f48803f3c9ba8ba0e00b
在您的情况下，您需要在每个轴上使用 idx：
a[idx[:,0], idx[:, 1], idx[:, 2]].shape == (2,) # True
以上就是索引多维 numpy 数组中的多个元素的详细内容，更多请关注php中文网其它相关文章！
每个人都需要一台速度更快、更稳定的 PC。随着时间的推移，垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是，许多工具可以让 Windows 保持平稳运行。
广告![ThinkPHP5快速开发企业站点[全程实录]](https://img.php.cn/upload/course/000/000/068/6253d918a3ce7278.png)
收藏
收藏
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号
494
