考虑到形状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),一个每个维度有一个值的元组?