从多级字典中提取特定路径

Question

我有一个矩阵:

matrix = {
    'A' : {
        'A1' : { 
            'A11' : [1,2,3],
            'A12' : [4,5,6],
        },
        'A2' : {
            'A21' : [11,12,14],
            'A22' : [14,15,16],
        },
        'A3' : {
            'A31' : [111,112,114],
            'A32' : [114,115,116],
        },
    }
}

我想检索动态查询的特定路径 - 例如matrix['A']['A2']['A22']或matrix['A']['A2'].

简单来说,我有一个多级字典,以及映射到该字典中层次结构的字符串列表.如['A','A1','A2']

我不确定最狡猾的做法.

以下作品.我只是想知道是否有更简洁或可读的方式.我喜欢另一双眼睛提供输入并纠正一个明显的错误.

get_target_path( pth , mtx ):
    try:
        value = mtx    
        for level in pth :
            value = value[level]
        return value
    except KeyError :
        return None

target_path = ['A','A2','A22']
result = get_target_path( target_path , matrix )

Answer 1

更短的选择是:

import operator
from functools import reduce  # in python 3 only; reduce is a builtin in python 2

def get_target_path(path, matrix):
    try:
        return reduce(operator.getitem, path, matrix)
    except KeyError:
        return None

但我认为它没有任何好转; 我可能会按照你的方式编写它(虽然间距不同:p).