2
我想索引一个特定的numpy数组项目的邻居。例如,如果我具有以下所示的阵列和我检查项目超过一定值我怎样才能索引单元的上方,下方,左侧和以有效的方式真实项目的权利,而不诉诸循环等。有效索引numpy数组邻居
In [34]: x
Out[34]:
array([[ 10., 10., 10., 10., 10.],
[ 10., 10., 10., 10., 10.],
[ 10., 20., 10., 10., 10.],
[ 10., 10., 10., 20., 10.],
[ 10., 10., 10., 10., 10.]])
In [37]: ans = x > 10
In [38]: ans
Out[38]:
array([[False, False, False, False, False],
[False, False, False, False, False],
[False, True, False, False, False],
[False, False, False, True, False],
[False, False, False, False, False]], dtype=bool)