2012-02-16 76 views
13

我期待在python中创建一个简单的“查找”机制,并希望确保没有什么东西隐藏在python的庞大的图书馆中,尚未做到这在创建之前。访问python字典与多个键查找字符串

我期待采取格式化像这样

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

的字典,我想有办法通过使用十进制表示,这将是类似的东西,以

访问数据
root.secondary.user2 

并将返回结果字典作为回应。我在想,一定有这样做,我可以写一个没有太大困难,但我想确保我没有重新创建我可能从文档中丢失的东西。由于

+0

你在寻找语法糖('mydict.root.secondary.user2')还是只通过字符串查找? – 2012-02-16 23:40:44

回答

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]] 
+0

这适用于检索,但关于实际更新该字典中的字段的任何想法使用相同的方法 – Tanerax 2012-02-16 23:20:34

+0

+1作为reduce()的答案;尽管我也很简单,就像答案一样,希望我能+2。 – SingleNegationElimination 2012-02-16 23:39:22

+0

@Tanerax:要进行更新,请拆分密钥的最后一个组件,并使用主要组件检索包含的字典。为答案添加了示例代码。 – 2012-02-16 23:41:37

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]可用于获取或投入新的价值。

2

你可以有。您可以通过使用类似于下面的代码来继承字典,添加密钥查找(甚至保留名称字典)。然而,{...}表单仍将使用内置的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