python - 按照列表 A 的排序对列表 B 进行排序？

Question

假设我有一些包含 k 个元素的列表 A，以及包含 k 个元素的列表 B。我想对列表 A 进行排序，但我也想以相同的方式排列列表 B。

例如

A = [2,3,1,4]
B = [5,6,7,8]

排序A后:

A = [1,2,3,4]
B = [7,5,6,8]

Answer 1

这是一种方法:

>>> A = [2,3,1,4]
>>> B = [5,6,7,8]
>>> A, B = zip(*sorted(zip(A, B)))
>>> list(A)
[1, 2, 3, 4]
>>> list(B)
[7, 5, 6, 8]

简而言之:

• 将A和B压缩成一对列表；
• 对配对进行排序；
• unzip回到A和B；
• 将元组转换为列表。

如果你喜欢单行本:

A, B = map(list, zip(*sorted(zip(A, B))))