对A列表进行排序，排序规则为A元素在B列表出现的顺序

A = [1,6,3,5,2]
B = [3,2,5,1,6]

A.sort(key=lambda elem:B.index(elem))
print(A) #[3, 2, 5, 1, 6]



a = [
    [1,2,3],
    [2,3,4],
    [4,2,3],
    [3,4,5],
    [5,5,6],
    [4,5,6],
]
b = [5,1,2,4,3]

a.sort(key=lambda elem:b.index(elem[0]))
print(a)   # [[5, 5, 6], [1, 2, 3], [2, 3, 4], [4, 2, 3], [4, 5, 6], [3, 4, 5]]