使用多个键查找字符串访问python dict

时间:2012-02-16 22:50:38

标签: python dictionary

我希望在python中创建一个简单的“查找”机制,并希望确保在创建它之前还没有在python中的大型库中隐藏某些内容尚未执行此操作的内容。

我正在寻找一个像这样格式化的字典

my_dict = { 
  "root": { 
    "secondary": { 
      "user1": { 
          "name": "jim", 
          "age": 24 
      }, 
      "user2": { 
        "name": "fred", 
        "age": 25 
      } 
    } 
  } 
}

我试图通过使用类似于

的十进制表示法来访问数据
root.secondary.user2

并将返回的结果作为回复返回。我认为必须有一些东西可以做到这一点,我可以毫不费力地写一个,但我想确保我没有重新创建我可能在文档中遗漏的东西。感谢

4 个答案:

答案 0 :(得分:28)

标准库中没有任何内容可用于此目的,但您自己编写代码非常容易:

>>> key = "root.secondary.user2"
>>> reduce(dict.get, key.split("."), my_dict)
{'age': 25, 'name': 'fred'}

这利用了以下事实:字典k中的密钥d的查找可以写为dict.get(d, k)。使用reduce()迭代地应用此结果会产生所需的结果。

编辑:为了完整性,使用此方法获取,设置或删除字典键的三个函数:

def get_key(my_dict, key):
    return reduce(dict.get, key.split("."), my_dict)

def set_key(my_dict, key, value):
    key = key.split(".")
    my_dict = reduce(dict.get, key[:-1], my_dict)
    my_dict[key[-1]] = value

def del_key(my_dict, key):
    key = key.split(".")
    my_dict = reduce(dict.get, key[:-1], my_dict)
    del my_dict[key[-1]]

答案 1 :(得分:2)

你可以拥有它。您可以使用类似于下面的代码来继承dict,添加键查找(甚至保留名称dict)。然而,{...}形式仍将使用内置dict类(现在称为orig_dict),因此您必须将其括起来,如下所示:Dict({...})。此实现以递归方式将字典转换为新表单,因此您不必将上述方法用于任何纯字典本身的字典条目。

orig_dict = dict
class Dict(orig_dict):
    def __init__(self, *args, **kwargs):
        super(Dict, self).__init__(*args, **kwargs)
        for k, v in self.iteritems():
            if type(v) == orig_dict and not isinstance(v, Dict):
                super(Dict, self).__setitem__(k, Dict(v))
    def __getattribute__(self, k):
        try: return super(Dict, self).__getattribute__(k)
        except: return self.__getitem__(k)
    def __setattr__(self, k, v):
        if self.has_key(k): self.__setitem__(k, v)
        else: return super(Dict, self).__setattr__(k, v)
    def __delattr__(self, k):
        try: self.__delitem__(k)
        except: super(Dict, self).__delattr__(k)
    def __setitem__(self, k, v):
        toconvert = type(v) == orig_dict and not isinstance(v, Dict)
        super(Dict, self).__setitem__(k, Dict(v) if toconvert else v)

# dict = Dict  <-- you can even do this but I advise against it

# testing:
b = Dict(a=1, b=Dict(c=2, d=3))
c = Dict({'a': 1, 'b': {'c': 2, 'd': 3}})
d = Dict(a=1, b={'c': 2, 'd': {'e': 3, 'f': {'g': 4}}})

b.a = b.b
b.b = 1
d.b.d.f.g = 40
del d.b.d.e
d.b.c += d.b.d.f.g
c.b.c += c.a
del c.a
print b
print c
print d

答案 2 :(得分:1)

递归仍然有效。

def walk_into( dict, key ):
    head, _, tail = key.partition('.')
    if tail:
        return walk_into( dict[head], tail )
    return dict, key
d, k = walk_into( my_dict, "root.secondary.user2" )

d[k]可用于获取或添加新值。

答案 3 :(得分:0)

我有一个非常完整的实现和其他一些东西here。如果您不想安装它,存储库 heretrict.util__get__ 中的 trict.trict 方法相结合可能有您需要的东西。此外,它实际上是在 conda-forge 中,即使 README 可能会另有说明,如果在您阅读本文之前我还没有来得及更新它。

相关问题