思路:定义一个类,包括数据和索引,这样排序之后都能找到对应的索引了
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
网友评论