得到numpy.ndarray的索引

Question

我有一个numpy.ndarray类型的一维数组,我想知道它的最大条目的索引.找到最大值后,我用了

peakIndex = numpy.where(myArray==max)

找到峰值指数.但是我的脚本吐出来而不是索引

peakIndex = (array([1293]),)

我希望我的代码只吐出整数1293.如何清理输出？

Answer 1

而不是使用numpy.where,你可以使用numpy.argmax.

peakIndex = numpy.argmax(myArray)

numpy.argmax返回单个数字,第一次出现的最大值的展平索引.如果myArray是多维的,您可能希望将展平的索引转换为索引元组:

peakIndexTuple = numpy.unravel_index(numpy.argmax(myArray), myArray.shape)