[解決済み] キーのリストを介してネストされた辞書アイテムにアクセスしますか？

質問

複雑な辞書構造を持っており、キーのリストを介してアクセスし、正しい項目を指定したいと考えています。

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 list
def getFromDict(dataDict, mapList):    
    for k in mapList: dataDict = dataDict[k]
    return dataDict

# Set a given data in a dictionary with position provided as a list
def setInDict(dataDict, mapList, value): 
    for k in mapList[:-1]: dataDict = dataDict[k]
    dataDict[mapList[-1]] = value

解決方法は？

使用方法 reduce() で辞書をたどる。

from functools import reduce  # forward compatibility for Python 3
import operator

def 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}}

Python PEP8 スタイルガイドの 関数にsnake_case名を指定します。 . 上記はリストや、辞書とリストの混在したものに対しても同様に機能します。 get_by_path() と set_by_path() :

from functools import reduce  # forward compatibility for Python 3
import operator

def 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

そして、補完のために、キーを削除する関数。

def del_by_path(root, items):
    """Delete a key-value in a nested object in root by item sequence."""
    del get_by_path(root, items[:-1])[items[-1]]