按列表A排序列表B?

时间:2015-01-07 20:10:43

标签: python list

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

2 个答案:

答案 0 :(得分:10)

这是一种方式:

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

简而言之:

  • zip AB成对列表;
  • 对这些对进行排序;
  • unzip返回AB;
  • 将元组转换为列表。

如果你喜欢单行:

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

答案 1 :(得分:2)

您可以尝试这样的事情:

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

我们在这里所做的就是“压缩”列表(例如在您的示例中:[(2, 5), (3, 6), (1, 7), (4, 8)])并按每个元组的第一个元素排序 列表。然后,我们会从此排序列表中检索所需的AB