通过Python中的函数访问字典的可变数量的键

Question

此功能适用于两个键：

def get(keyA, keyB)
    return mydict[keyA][keyB]

如何为可变数量的键编写以下函数？

def get(*keys)
    return mydict[keys[0]][keys[1]][keys[3]]...

Answer 1

迭代键，并在迭代时分配项目：

def get(*keys):
    ret = mydict
    for key in keys:
        ret = ret[key]
    return ret