相关疑难解决方法(0)

获取ndarray中N个最高值的索引

考虑到形状100x100x100的直方图,我想找到2个最高值a和b,以及它们的索引(a1,a2,a3)和(b1,b2,b3),例如:

hist[a1][a2][a3] = a
hist[b1][b2][b3] = b
Run Code Online (Sandbox Code Playgroud)

我们可以使用hist.max()轻松获得最高值,但是如何在ndarray中获得X最高值?

我知道通常使用np.argmax来检索值索引,但在这种情况下:

hist.argmax().shape = ()  # single value
for i in range(3):
    hist.argmax(i).shape = (100, 100)
Run Code Online (Sandbox Code Playgroud)

我怎样才能得到一个形状(3),一个每个维度有一个值的元组?

python indexing numpy multidimensional-array

6
推荐指数
1
解决办法
2638
查看次数

标签 统计

indexing ×1

multidimensional-array ×1

numpy ×1

python ×1