我正在尝试获取Numpy数组中最大元素的索引。这可以使用numpy.argmax
来完成。我的问题是,我想要找到整个数组中最大的元素,并获得它的索引。
numpy.argmax
既可以沿着一个轴应用,这不是我想要的,也可以应用在扁平化的数组上,这是我想要的。
我的问题是,当我需要多维索引时,结合使用numpy.argmax
和axis=None
会返回平面索引。
我可以使用divmod
来获得非平坦的索引,但这感觉很难看。有没有更好的方法来做这件事?
https://stackoverflow.com/questions/9482550
复制相似问题