Variable number of keys to access dictionary through a function in Python

Question

This function is for two keys:

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

How can I write the following function for variable number of keys?

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

And presumably you did not mean to skip `keys[2]` there. – Martijn Pieters Dec 08 '14 at 14:06 — Martijn Pieters, Dec 08 '14 at 14:06

score 1 · Accepted Answer · answered Dec 08 '14 at 14:05

1

Iterate keys, and assign the item as you iterate:

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

answered Dec 08 '14 at 14:05

falsetru

1 Answers1