Python排序进阶版：根据一个列表的顺序对其他列表进行排序

Posted 2022-10-15 大彤小忆

  在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]