按照列表 A 的排序对列表 B 进行排序?

Sort list B by the sort of list A?

假设我有一些包含 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]

您可以尝试这样的操作:

>>> A = [2,3,1,4]
>>> B = [5,6,7,8]
>>>
>>> AB = zip(A, B)
>>> AB.sort()
>>> A[:] = [t[0] for t in AB]
>>> B[:] = [t[1] for t in AB]
>>> A
[1, 2, 3, 4]
>>> B
[7, 5, 6, 8]

我们在这里所做的只是 "zipping" 列表(即在您的示例中:[(2, 5), (3, 6), (1, 7), (4, 8)])并按 that 列表的第一个元素排序每个元组。然后从这个排序列表中我们检索所需的 AB.

这是一种方法:

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

简而言之:

  • AB 压缩成一对列表;
  • 对成对排序;
  • unzip 回到 AB;
  • 将元组转换为列表。

如果你喜欢单行本:

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