python:检索字典或集合中的天花板键和地板键

Question

我有一本字典，里面有一堆整数键。对于我没有的键，我希望能够在我想检索但不存在的键之前和之后检索最小和最大的键。
Java 中的 Treemap 类有两个方法可以做到这一点:ceilingkey() 和 floorkey()。

How can I do this with python?

举个例子，我有一个这样的字典:

 { 1: "1", 4: "4", 6: "6" ..., 100: "100" } 

如果我要求输入 key 1，我将检索 "1"，但是如果我寻找键 3，我应该得到 KeyError，因此能够得到 floor(3) = 1 和 ceil (3) = 4。

Answer 1

def floor_key(d, key):
    if key in d:
        return key
    return max(k for k in d if k < key)

def ceil_key(d, key):
    if key in d:
        return key
    return min(k for k in d if k > key)

我不确定您想如何处理边界条件。请注意，如果您要求的键的下限/上限低于/高于字典中的任何内容，这将引发异常(ValueError)。