Python 中的 3 维 Numpy ndarray 中的高级索引

Question

我有一个形状为 (68, 64, 64) 的数组 'prediction'。这些尺寸对应 image_number、高度、宽度。对于每个图像，我都有一个长度为 2 的元组，其中包含对应于每个 64x64 图像中特定位置的坐标，例如 (12, 45)。我可以将这些坐标堆叠到另一个形状为 (68,2) 的 Numpy ndarray 中，称为 'locations'.

如何在不使用循环的情况下构造切片对象或构造必要的高级索引来访问这些位置？寻求有关语法的帮助。使用没有循环的纯 Numpy 矩阵是目标。

工作循环结构

Import numpy as np
# example code with just ones...The real arrays have 'real' data.
prediction = np.ones((68,64,64), dtype='float32')
locations = np.ones((68,2), dtype='uint32')

selected_location_values = np.empty(prediction.shape[0], dtype='float32')
for index, (image, coordinates) in enumerate(zip(prediction, locations)):
    selected_locations_values[index] = image[coordinates]

期望的方法

selected_location_values = np.empty(prediction.shape[0], dtype='float32')

correct_indexing = some_function_here(locations). # ?????
selected_locations_values = predictions[correct_indexing]

Answer 1

一个简单的索引应该可以工作：

img = np.arange(locations.shape[0])
r = locations[:, 0]
c = locations[:, 1]
selected_locations_values = predictions[img, r, c]

花式索引通过选择索引数组中与广播索引的形状相对应的元素来工作。在这种情况下，索引非常简单。你只需要范围就可以告诉你每个位置对应的图像。

Python 中的 3 维 Numpy ndarray 中的高级索引

Advanced Indexing in 3 Dimensional Numpy ndarray In Python

python

numpy

numpy-indexing

array-broadcasting