Python列表直接访问dict

Question

我有一个这样的文档：

>>> k = {'finance_pl':{'S':{'2008':45,'2009':34}}}

正常的访问方式是：

>>> k['finance_pl']['S']
{'2008': 45, '2009': 34}

但是，在我的情况下，最终用户会以finance_pl.S 的形式给我输入

我可以将其拆分并访问字典，如下所示：

>>> doc_list = doc.split('.')
>>> k[doc_list[0]][doc_list[1]]
{'2008': 45, '2009': 34}

但是，我不想这样做，因为字典结构可能会改变和用户可以给出类似于CCD_ 2而不是CCD_ 3或CCD_。

我需要一些东西来直接应用用户输入（例如：如果输入是finance_pl.new.S，我应该能够将这个.split('.')方法应用于用户输入并直接应用）。

做这件事的优雅方法是什么？

Answer 1

我会简单地循环所有部分：

def getter(somedict, key):
    parts = key.split(".")
    for part in parts:
        somedict = somedict[part]
    return somedict

之后我们有

>>> getter(k, "finance_pl.S")
{'2008': 45, '2009': 34}

或

>>> getter({"a": {"b": {"c": "d"}}}, "a")
{'b': {'c': 'd'}}
>>> getter({"a": {"b": {"c": "d"}}}, "a.b.c")
'd'

Answer 2

您可以选择以下内容：

k = {'finance_pl':{'S':{'2008':45,'2009':34}}}
print reduce(dict.__getitem__, 'finance_pl.S.2009'.split('.'), k)
# 34

如果你使用的是Python3.x，你需要一个from functools import reduce。。。

Answer 3

>>> k = {'finance_pl':{'S':{'2008':45,'2009':34}}}
>>> ui = 'finance_pl.S'
>>> def getter(adict, key):
...     return reduce(dict.get, key.split('.'), adict)
... 
>>> getter(k, ui)
{'2008': 45, '2009': 34}
>>> 

Answer 4

userkey = 'finance_pl.new.S'
for key in userkey.split('.'):
    k = k[key]
# the final value of k is the one you want

因此，基本上只需迭代每个子键并检索内部字典，直到您用完子键