python - 根据具有特殊要求的另一个数组对数组进行排序

Question

我有一个像这样的数组:

master = ['aa','bb','cc','dd','ee','ff']

我需要按照这个排序:

order = ['c','e','a','b']

所以最终的答案如下:

['cc','ee','aa','bb','dd','ff']

问题是，如果该项目不在订单数组中，那么它会走到最后，我不知道该怎么做。

你会如何处理这个问题？我尝试过:

master.sort(key=lambda x: order.index(x))

但是如果x不在order数组中，例如'dd'和'ff'

Answer 1

您可以使用maketrans和 translate :

master = ['aa','bb','cc','dd','ee','ff']
order = 'ceab'
trantab = str.maketrans(order, 'abcdefghijklmnopqrstuvwxyz'[:len(order)])
master.sort(key=lambda x: x.translate(trantab))

根据您对“fecd”等顺序值的期望，您还可以按如下方式创建转换表:

trantab = str.maketrans(order, ''.join(sorted(order)))

...这将使“a”和“b”排在“f”之前。