提取包含特定值的数组索引

时间:2016-07-03 05:00:38

标签: python arrays numpy

我有一个形状如下的数组:(31777,44,44)

sum为31777.0,每个(44,44)网格在某些坐标处只包含一个1.0条目。

我的目标是获得一个形状(31777,2)的数组,其中2是网格中1.0条目的坐标。 执行此操作的混乱方法是使用循环,但我正在寻找一种优雅的高性能解决方案。

2 个答案:

答案 0 :(得分:0)

您可以使用col-xs-10查找多维数组中元素的索引,例如:

numpy.where

答案 1 :(得分:0)

第一个答案给了我谷歌材料,找到我认为最佳的材料:

np.vstack(arr.nonzero()[1:][::-1]).T