CODEWITHSUNDEEP
pythonfunctionarguments
hengyue li
hengyue li

Reputation: 468

Python: How to only pass needed arguments into function from a dictionary?

Assume now I have a well-defined function

def f1(a):
    ...

and i can not change its definition. Now I want to call it with a argument from a dictionary 

D={ 'a':1 , 'b':2}

If I call it like

f1(**D)

there will be a syntax error (because b is not defined in f1). My question is if I can find a smart way to let the function to find the argument it needed only?

Upvotes: 2

Views: 1274

Answers (3)

Joe Iddon
Joe Iddon

Reputation: 20424

You can use inspect.getargspec:

d = {'a': 5, 'b': 6, 'c': 7}
def f(a,b):
    return a + b

f(*[d[arg] for arg in inspect.getargspec(f).args])

which gives 11.

Thanks Eugene, getargspec is legacy, instead you should use getfullargspec which has the same usage in this case.

Upvotes: 9

Eugene Yarmash
Eugene Yarmash

Reputation: 149804

You could get the argument names from the function's code object, then lookup the values for them in the dictionary:

>>> def f(a):
...     b = 2
...     return a + b
...
>>> D = {'a': 1, 'b': 2, 'c': 3}
>>> nargs = f.__code__.co_argcount
>>> varnames = f.__code__.co_varnames
>>> f(*(D[arg] for arg in varnames[:nargs]))
3

Upvotes: 1

chenchuk
chenchuk

Reputation: 5742

Here is an example of function that got a dictionary and use only the 'b' key :

In [14]: def f1(mydict):
    ...:     return mydict['b']
    ...: 

In [15]: x={'a': 1, 'b':2}

In [16]: f1(x)
Out[16]: 2

Upvotes: -3

Related Questions