返回非平面索引的numpy数组的Argmax
返回非平面索引的numpy数组的Argmax
提问于 2012-02-28 21:12:05
我正在尝试获取Numpy数组中最大元素的索引。这可以使用numpy.argmax来完成。我的问题是，我想要找到整个数组中最大的元素，并获得它的索引。
numpy.argmax既可以沿着一个轴应用，这不是我想要的，也可以应用在扁平化的数组上，这是我想要的。
我的问题是，当我需要多维索引时，结合使用numpy.argmax和axis=None会返回平面索引。
我可以使用divmod来获得非平坦的索引，但这感觉很难看。有没有更好的方法来做这件事？
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/9482550
复制相关文章
相似问题