Python 使用键列表获取字典值

Question

考虑到我有

users = \
{
    'Sam': {
        'age': 19,
        'location': 'Flodira',
        'country': 'United States'
    },
    'Max': {
        'age': 16,
        'location': 'Sydney',
        'country': 'Australia'
    }
}

def getUserValue(query):
    return users[query]

getUserValue(['Sam', 'location'])

我该怎么做？我知道我可以直接执行 users['Sam']['location']，我想做的是通过列表对象获取 Sam 的位置。

Answer 1

这是处理不同级别嵌套的通用解决方案:

def getUserValue(users, query):
    v = users
    for i in query:
        v = v[i]
    return v

print(getUserValue(users, ['Sam', 'location']))
# Flodira