排序字典键的子集

Question

如何根据A的值对以下列表B进行排序,该列表是字典A的子集？

A = {1: 10, 2:20, 3: 15, 4: 1}
B = [1, 3, 4]

我的解决方案应该是

B = [4, 1, 3]

以来

A[4] < A[1] < A[3]

Answer 1

使用A.get如key

A = {1: 10, 2:20, 3: 15, 4: 1}
B = [1, 3, 4]

>>> sorted(B, key=A.get)
[4, 1, 3]

请注意,您可以将默认值设置为.get方法的参数,这样,如果您的dict中不存在给定的键,您可以自己设置位置(例如,最后一个位置或第一个位置).例如:

B = [1, 3, 4, 11]
>>> sorted(B, key=lambda k: A.get(k, -1))
[11, 4, 1, 3]

>>> sorted(B, key=lambda k: A.get(k, float('inf')))
[4, 1, 3, 11]

或者,正如@jpp强调的那样,

>>> sorted(B, key=A.__getitem__)

如果您希望KeyError在引用键不存在的情况下引发它.