如何根据一个列表重新排序两个不同大小的列表？

Question

给定两个共享元素但大小不同的不同列表，如何根据第一个列表的元素顺序对第二个列表重新排序？例如：

为了：

a = ['a', 'b', 'e', 'z', 'f']
b = ['a', 'f', 'b', 'z', 'g', 'h']

如何将列表 b 重新排序为：

['a', 'b', 'z', 'f', 'g', 'h']

为了：

a = ['a', 'c', 'j', 'r' , 'p']
b = ['b', 'c', 'a']

如何将列表 b 重新排序为：

['a', 'c', 'b']

到目前为止，我试图：

[x for y, x in sorted(zip(a, b))]

但是，我不明白如何控制列表大小不同的事实。我如何根据上述限制重新排序第二个列表？

Answer 1

您可以提供一个键函数来根据索引进行排序。

a = ['a', 'b', 'e', 'z', 'f']
b = ['a', 'f', 'b', 'z', 'g', 'h']
from math import inf
res = sorted(b, key=lambda x: a.index(x) if x in a else inf)
print(res)