同时排序多个列表

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