同时排序多个列表

Question

我有三个由相同数量的元素组成的列表：

a = [0.3, 1.5, 0.2, 2.6]
b = [1, 2, 3, 4]
c = [0.01, 0.02, 0.03, 0.04]

我需要根据列表a中的减少值同时对这三个列表进行排序。这意味着我必须根据a排序，但a也需要排序。 输出应如下所示：

a_s = [2.6, 1.5, 0.3, 0.2]
b_s = [4, 2, 1, 3]
c_s = [0.04, 0.02, 0.01, 0.03]

这种方式a_s现在只需按值减少排序，而b_s和c_s根据此排序更改其项目位置。

Answer 1

简短回答

a_s, b_s, c_s = map(list, zip(*sorted(zip(a, b, c), reverse=True)))

长答案

首先，您必须压缩三个列表，创建一个项目列表：

>>> a = [0.3, 1.5, 0.2, 2.6]
>>> b = [1, 2, 3, 4]
>>> c = [0.01, 0.02, 0.03, 0.04]
>>> z = zip(a, b, c)
>>> z
[(0.3, 1, 0.01), (1.5, 2, 0.02), (0.2, 3, 0.03), (2.6, 4, 0.04)]

然后，您对此列表进行排序。元组列表按其第一个元素排序（当第一个元素相等，第二个元素使用时，依此类推）：

>>> zs = sorted(z, reverse=True)
>>> zs
[(2.6, 4, 0.04), (1.5, 2, 0.02), (0.3, 1, 0.01), (0.2, 3, 0.03)]

然后你“解压缩”列表。解压缩与使用每个元组作为参数调用zip相同，这是通过使用星形语法实现的：

>>> u = zip(*zs)
>>> u
[(2.6, 1.5, 0.3, 0.2), (4, 2, 1, 3), (0.04, 0.02, 0.01, 0.03)]

你得到一个元组列表，但你想要列表。因此，您将列表构造函数映射到以下项目：

>>> u
[(2.6, 1.5, 0.3, 0.2), (4, 2, 1, 3), (0.04, 0.02, 0.01, 0.03)]
>>> map(list, u)
[[2.6, 1.5, 0.3, 0.2], [4, 2, 1, 3], [0.04, 0.02, 0.01, 0.03]]

然后，您可以将列表解压缩为变量：

>>> a_s, b_s, c_s = map(list, u)

观察

排序时，您可以明确指出将用于排序的项目，而不是依赖于元组的默认排序：

>>> from operator import itemgetter
>>> sorted(z, key=itemgetter(1))  # Sort by second item
[(0.3, 1, 0.01), (1.5, 2, 0.02), (0.2, 3, 0.03), (2.6, 4, 0.04)]
>>> sorted(z, key=itemgetter(2))  # Sort by third item
[(0.3, 1, 0.01), (1.5, 2, 0.02), (0.2, 3, 0.03), (2.6, 4, 0.04)]