python - 根据并行列表中的相应值对列表进行排序

Question

这个问题已经有答案了:

Given parallel lists, how can I sort one while permuting (rearranging) the other in the same way? (16 个回答)

已关闭 5 个月前。

我有一个像这样的字符串列表:

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构造，但我很好奇是否有更短的方法。有什么建议吗？

Answer 1

最短代码

[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. zip两个列表。
2. 使用sorted()基于zip创建一个新的、排序的列表 .
3. 使用列表推导式从已排序、压缩的列表中提取每对的第一个元素。

有关如何设置\使用 key 参数以及一般 sorted 函数的更多信息，请查看 this .

<小时/>