如何在 NumPy 中使用整数数组索引从多维数组中提取元素？

使用整数数组索引提取元素

使用多维数组时，通常需要根据索引提取特定元素。在 NumPy 中，您可以使用各种技术来实现此目的。其中一种方法是使用整数数组索引。

考虑以下示例：

我们的目标是创建一个新数组 C，其中包含 A 中的元素，其中每个元素的行索引元素由 A.shape[0] 给出，列索引由 B 的 raveled 版本给出。换句话说，C 应该是：

一种方法是使用整数数组索引，如下所示：

此方法使用 arange 函数为 A 的行生成一系列索引，然后将其与 B 的 raveled 版本组合以创建列索引。结果是一个包含所需元素的新数组。

输出：

需要注意的是，如果 B 是一维数组或列表，您可以使用 .ravel() 跳过展平操作。

以上是如何在 NumPy 中使用整数数组索引从多维数组中提取元素？的详细内容。更多信息请关注PHP中文网其他相关文章！