【Python】sorted排序并返回索引

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

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