通过键列表访问嵌套字典项？

Question

通过键列表访问嵌套字典项？我有一个复杂的字典结构，我想通过一个键列表来访问它，以找到正确的项。dataDict = {     "a":{         "r": 1,         "s": 2,         "t": 3         },     "b":{         "u": 1,         "v": {             "x": 1,             "y": 2,             "z": 3         },         "w": 3         }}    maplist = ["a", "r"]或maplist = ["b", "v", "y"]我已经做了下面的代码，但我相信如果有人有想法的话，有一种更好、更有效的方法来做到这一点。# Get a given data from a dictionary with position provided as a listdef getFromDict(dataDict, mapList):         for k in mapList: dataDict = dataDict[k]     return dataDict# Set a given data in a dictionary with position provided as a listdef setInDict(dataDict, mapList, value):      for k in mapList[:-1]: dataDict = dataDict[k]     dataDict[mapList[-1]] = value

Answer 1

使用reduce()翻阅字典：

from functools import reduce  # forward compatibility for Python 3import operatordef getFromDict(dataDict, mapList):
    return reduce(operator.getitem, mapList, dataDict)

再利用getFromDict若要查找要存储值的位置，请执行以下操作setInDict():

def setInDict(dataDict, mapList, value):
    getFromDict(dataDict, mapList[:-1])[mapList[-1]] = value

除了最后一个元素mapList需要找到要添加值的“父”字典，然后使用最后一个元素将值设置为正确的键。

演示：

>>> getFromDict(dataDict, ["a", "r"])1>>> getFromDict(dataDict, ["b", "v", "y"])2>>> setInDict(dataDict, ["b", "v", "w"], 4)>>> 
import pprint>>> pprint.pprint(dataDict){'a': {'r': 1, 's': 2, 't': 3},
 'b': {'u': 1, 'v': {'w': 4, 'x': 1, 'y': 2, 'z': 3}, 'w': 3}}

注意，PythonPEP 8样式指南为函数规定了Snake_case名称..上面的方法同样适用于列表或字典和列表的混合，所以名称应该是get_by_path()和set_by_path():

from functools import reduce  # forward compatibility for Python 3import operatordef get_by_path(root, items):
    """Access a nested object in root by item sequence."""
    return reduce(operator.getitem, items, root)def set_by_path(root, items, value):
    """Set a value in a nested object in root by item sequence."""
    get_by_path(root, items[:-1])[items[-1]] = value

Answer 2

1. 接受的解决方案不会直接对python 3起作用-它将需要一个

   from functools import reduce.

2. 而且，使用一个

   for

   循环。见引用自

   Python3.0的新特性是什么？.

   移除reduce()..使用functools.reduce()如果您真的需要它，但是99%的时间是显式的for循环更易读。

3. 接下来，接受的解决方案不会设置不存在的嵌套键(它返回

   KeyError

   )-见@EAFIT的答案

那么，为什么不使用Kolergy问题中建议的方法来获得一个值：

def getFromDict(dataDict, mapList):    
    for k in mapList: dataDict = dataDict[k]
    return dataDict

以及@EAFIT用于设置值的答案中的代码：

def nested_set(dic, keys, value):
    for key in keys[:-1]:
        dic = dic.setdefault(key, {})
    dic[keys[-1]] = value

两者都直接在python 2和3中工作。

Answer 3

使用REPLE很聪明，但是如果嵌套字典中不存在父键，OP的SET方法可能会出现问题。因为这是我在Google搜索中看到的第一篇关于这个主题的帖子，我想让它稍微好一点。

中的集合方法在嵌套python字典中设置一个值，给出索引和值的列表)似乎对丢失的父母钥匙更有信心。复制它：

def nested_set(dic, keys, value):
    for key in keys[:-1]:
        dic = dic.setdefault(key, {})
    dic[keys[-1]] = value

此外，有一个遍历密钥树并获取所有绝对密钥路径的方法也很方便，我已经为这些路径创建了如下内容：

def keysInDict(dataDict, parent=[]):
    if not isinstance(dataDict, dict):
        return [tuple(parent)]
    else:
        return reduce(list.__add__, 
            [keysInDict(v,parent+[k]) for k,v in dataDict.items()], [])

它的一种用途是使用以下代码将嵌套树转换为熊猫DataFrame(假设嵌套字典中的所有LEAF具有相同的深度)。

def dict_to_df(dataDict):
    ret = []
    for k in keysInDict(dataDict):
        v = np.array( getFromDict(dataDict, k), )
        v = pd.DataFrame(v)
        v.columns = pd.MultiIndex.from_product(list(k) + [v.columns])
        ret.append(v)
    return reduce(pd.DataFrame.join, ret)