python - 选择二维 NumPy 数组的 "corner"个元素

Question

这个问题在这里已经有了答案:

Get corner values in Python numpy ndarray (4 个回答)

关闭 4年前。

我正在尝试提取 NumPy 二维数组的四个角元素:

import numpy as np
data = np.arange(16).reshape((4, -1))
#array([[ 0,  1,  2,  3],
#       [ 4,  5,  6,  7],
#       [ 8,  9, 10, 11],
#       [12, 13, 14, 15]])

预期的输出是 [[0,3],[12,15]] 或 [0,3,12,15] (什么都可以)。真正的 2D fancy 索引只提供主对角线的末端:

data[[0,-1],[0,-1]]
#array([ 0, 15])

伪二维 fancy 索引(首先按行，然后按列)提供了正确的答案，但看起来很尴尬:

data[[0,-1]][:,[0,-1]]
#array([[ 0,  3],
#       [12, 15]])

有没有办法使用真正的fancy索引，比如data[XXX,YYY]，其中XXX和YYY 是列表/数组/切片，提取所有四个角？

Answer 1

你可以这样做:

data[[0, 0, -1, -1], [0, -1, 0, -1]]