我有一个表示距离的数组,因此我尝试使用3个较小且距离增加的索引来获取数组:
mvn
有什么想法吗?提前谢谢。
答案 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]