Python求一维数组中最大的几个数的索引（深度学习中的SoftMax可能会用到）

import heapq

re = map(list(res).index, heapq.nlargest(5, res))  # 求最大的五个索引    nsmallest与nlargest相反，求最小

这里res是输入的一维数组，对长度没有要求。

nlargest(5, res)中的5表示求前五个最大的。如果需要求前6个，改为6即可，其他以此类推。