# 以数组 a 的从小到大的顺序为基准，对数组b进行重排序，并返回排序结果的索引数

import numpy as np

b = np.array([0,1,2,3,4,5,6])

index = np.lexsort((b, a))