获取指定索引处的数组元素

Question

我有两个 Numpy 数组：

B：形状：（20、20、4、3、4）

S：形状：（20、20、4、3）

这两个 arrays 是对应的，所以对于 S 中的每个元素，都存在一个 Array，其中 B 中的 4 个元素。

我使用 S 上的 argmax 函数来获取最大值的索引（例如 1929）

现在我的问题是如何使用给定的索引获取 B 中的数组？

当我调用 B.item(index) 时，我只得到一个元素，而不是数组。

Answer 1

默认情况下， argmax在展平数组上工作，因此您需要解开该索引以在另一个 NDarray 上使用它：

out = B[np.unravel_index(index, S.shape)]

展平的B数组的第 1929 个元素（这是B.item(index)给您的）只是一个值，因为它是展平的并且在该点没有 ND 结构。

Answer 2

尝试：

B[np.where(B == np.amax(B))]