在Python中如果需要根据列表A对列表B进行排序的问题时有以下2种方法:
- 方法1: 根据列表B中每个元素的下标来获取列表A中对应位置的元素,将其作为排序依据即可。
a = [1, 5, 2, 6, 7] b = [6, 3, 9, 5, 0] # sort list b by list a sorted_b = sorted(b, key=lambda x: a[b.index(x)]) print(sorted_b)
[6, 9, 3, 5, 0]
- 方法2: 用zip函数将两个列表打包,根据A排序,再解包即可。与方法1的区别在于方法2对A、B都进行了排序。
a = [1, 5, 2, 6, 7] b = [6, 3, 9, 5, 0] zip_a_b = zip(a, b) sorted_zip = sorted(zip_a_b, key=lambda x:x[0]) sorted_a, sorted_b = zip(*sorted_zip) print(list(sorted_a)) print(list(sorted_b))
[1, 2, 5, 6, 7]
[6, 9, 3, 5, 0]
第二种方法也可以根据列表A对列表B和C进行排序,如下所示。
a = [1, 5, 2, 6, 7] b = [6, 3, 9, 5, 0] c = [3, 1, 4, 5, 7] zip_a_b_c = zip(a, b, c) sorted_zip = sorted(zip_a_b_c, key=lambda x: x[0]) sorted_a, sorted_b, sorted_c = zip(*sorted_zip) print(list(sorted_a)) print(list(sorted_b)) print(list(sorted_c))
[1, 2, 5, 6, 7]
[6, 9, 3, 5, 0]
[3, 4, 1, 5, 7]