在Python中获取角落值numpy ndarray

Question

我试图访问numpy ndarray的角落值。对于方法论我绝对难以理解。任何帮助将不胜感激。

例如，从下面的数组我想要一个返回值的数组（[1,0,0,5]）或数组（[[1,0]，[0,5]]）。

array([[ 1., 0., 0., 0.], 
     [ 0., 1., 0., 0.], 
     [ 0., 0., 1., 5.], 
     [ 0., 0., 5., 5.]]) 

Answer 1

要添加各种问题的答案，就可以拿到角球项目的视图（不是副本）做：

corners = a[::a.shape[0]-1, ::a.shape[1]-1] 

或者，对于一般的n维数组：

corners = a[tuple(slice(None, None, j-1) for j in a.shape)] 

这样做，则可以通过修改视图修改原始数组：

>>> a = np.arange(9).reshape(3, 3) 
>>> a 
array([[0, 1, 2], 
     [3, 4, 5], 
     [6, 7, 8]]) 
>>> corners = a[tuple(slice(None, None, j-1) for j in a.shape)] 
>>> corners 
array([[0, 2], 
     [6, 8]]) 
>>> corners += 1 
>>> a 
array([[1, 1, 3], 
     [3, 4, 5], 
     [7, 7, 9]]) 

EDIT啊，你想要一个角落价值的平面列表......这通常不能用视图来实现，所以@ IanH的回答就是你要找的。

Answer 2

你可以手动指定角（使用负索引）：

a = numpy.array([[ 1., 0., 0., 0.], 
    [ 0., 1., 0., 0.], 
    [ 0., 0., 1., 5.], 
    [ 0., 0., 5., 5.]]) 

result = numpy.array([a[0][0],a[0][-1],a[-1][0],a[-1][-1]]) 
# result will contain array([ 1., 0., 0., 5.]) 

result = numpy.array([a[0][0],a[0][-1],a[-1][0],a[-1][-1]]) 
# result will contain array([[ 1., 0.], 
#       [ 0., 5.]]) 

Answer 3

如何

A[[0,0,-1,-1],[0,-1,0,-1]] 

其中A是数组。

Answer 4

使用np.ix_来构建指数。

>>> a 
array([[1., 0., 0., 0.], 
     [0., 1., 0., 0.], 
     [0., 0., 1., 5.], 
     [0., 0., 5., 5.]]) 

>>> corners = np.ix_((0,-1),(0,-1)) 

>>> a[corners] 
array([[1., 0.], 
     [0., 5.]])