基于其他列表中的值对列表进行排序

2024-03-29 07:06:12 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个字符串列表,如下所示:

X = ["a", "b", "c", "d", "e", "f", "g", "h", "i"]
Y = [ 0,   1,   1,   0,   1,   2,   2,   0,   1 ]

使用Y中的值对X进行排序以获得以下输出的最短方法是什么

["a", "d", "h", "b", "c", "e", "i", "f", "g"]

具有相同“键”的元素的顺序并不重要。我可以求助于使用for构造,但我很好奇是否有一种更短的方法。有什么建议吗


Tags: 方法字符串元素列表for排序顺序建议
3条回答

另外,如果您不介意使用numpy阵列(或者事实上已经在处理numpy阵列…),下面是另一个不错的解决方案:

people = ['Jim', 'Pam', 'Micheal', 'Dwight']
ages = [27, 25, 4, 9]

import numpy
people = numpy.array(people)
ages = numpy.array(ages)
inds = ages.argsort()
sortedPeople = people[inds]

我在这里找到的: http://scienceoss.com/sort-one-list-by-another-list/

将两个列表压缩在一起,进行排序,然后选择所需的部分:

>>> yx = zip(Y, X)
>>> yx
[(0, 'a'), (1, 'b'), (1, 'c'), (0, 'd'), (1, 'e'), (2, 'f'), (2, 'g'), (0, 'h'), (1, 'i')]
>>> yx.sort()
>>> yx
[(0, 'a'), (0, 'd'), (0, 'h'), (1, 'b'), (1, 'c'), (1, 'e'), (1, 'i'), (2, 'f'), (2, 'g')]
>>> x_sorted = [x for y, x in yx]
>>> x_sorted
['a', 'd', 'h', 'b', 'c', 'e', 'i', 'f', 'g']

将这些组合在一起,得到:

[x for y, x in sorted(zip(Y, X))]

最短代码

[x for _, x in sorted(zip(Y, X))]

示例:

X = ["a", "b", "c", "d", "e", "f", "g", "h", "i"]
Y = [ 0,   1,   1,    0,   1,   2,   2,   0,   1]

Z = [x for _,x in sorted(zip(Y,X))]
print(Z)  # ["a", "d", "h", "b", "c", "e", "i", "f", "g"]

一般来说

[x for _, x in sorted(zip(Y, X), key=lambda pair: pair[0])]

解释:

  1. ^{}两个{}
  2. 使用^{}基于zip创建一个新的、已排序的list
  3. 使用列表理解从已排序的压缩list中提取每对的第一个元素

有关如何设置\使用key参数以及sorted函数的更多信息,请参阅this


相关问题 更多 >