美文网首页
【Python】sorted排序并返回索引

【Python】sorted排序并返回索引

作者: 吵吵人 | 来源:发表于2020-08-22 23:43 被阅读0次

思路:定义一个类,包括数据和索引,这样排序之后都能找到对应的索引了

class dataset:
   def __init__(self, a, i):
       self.data = a
       self.ind = i

if __name__ == '__main__':
   m = 3
   A = [1, 5, 2, 8, 4, 9, 2, 6]
   arr = []
   for i in range(len(A)):
       arr.append(dataset(A[i], i))
   B = sorted(arr, key=lambda x: x.data)
   for i in range(m):
       print(B[i].ind, end=' ')

输出:

0 2 6
Process finished with exit code 0

相关文章

网友评论

      本文标题:【Python】sorted排序并返回索引

      本文链接:https://www.haomeiwen.com/subject/nfgbjktx.html