根据较长列表的顺序对短列表进行排序

Question

假设我有一个参考列表：

A = ['X_0', 'Z_0', 'X_1', 'Y_0', 'Z_1', 'X_2', 'Y_1', 'Z_2', 'Y_2']

以及一个（示例）较短的列表：

B = ['Z_0', 'X_1', 'X_0']

如何对B 进行排序，以使元素的顺序与A 中提供的顺序相匹配？因此，最终结果应该与A 的顺序相匹配，如下所示：

B_final = ['X_0', 'Z_0', 'X_1']

Answer 1

您希望列表 B 按 A 中值的索引排序，所以；

sorted(B, key=A.index)

这将按 A 中的每个值索引对 B 进行排序。

如果您使用的是一个很长的列表，索引效率不是很高，因为它会在列表中搜索每个值，因此您可能希望改为按字典排序。

首先创建列表索引值的字典；

>>> C = {val:ix for ix,val in enumerate(A)}

...然后按字典值排序，比在列表中线性搜索效率高很多；

>>> sorted(B, key=C.get)
['X_0', 'Z_0', 'X_1']