如何获得三个较小值的数组索引?

时间:2016-09-15 16:10:35

标签: arrays

我有一个表示距离的数组,因此我尝试使用3个较小且距离增加的索引来获取数组:

mvn

有什么想法吗?提前谢谢。

1 个答案:

答案 0 :(得分:0)

我找到了一种不太优雅但又有效的方法。如果有人有另一个快速或更好的版本,欢迎。

from operator import itemgetter
# b distances array
b = [i for i in enumerate(b)]
b = sorted(b, key=itemgetter(1))
b = [i[0] for i in b]
result = b[:3]
result
# [0,7,10]