我有一个形状如下的数组:(31777,44,44)
其sum
为31777.0,每个(44,44)
网格在某些坐标处只包含一个1.0条目。
我的目标是获得一个形状(31777,2)的数组,其中2
是网格中1.0条目的坐标。
执行此操作的混乱方法是使用循环,但我正在寻找一种优雅的高性能解决方案。
答案 0 :(得分:0)
您可以使用col-xs-10
查找多维数组中元素的索引,例如:
numpy.where
答案 1 :(得分:0)
第一个答案给了我谷歌材料,找到我认为最佳的材料:
np.vstack(arr.nonzero()[1:][::-1]).T