Я хотел бы вызвать функцию в Python, используя словарь.
Вот некоторый код:
d = dict(param='test')
def f(param):
print(param)
f(d)
Это печатает {'param': 'test'} но я бы хотел просто напечатать test.
Я хотел бы, чтобы он работал аналогично для других параметров:
d = dict(p1=1, p2=2)
def f2(p1, p2):
print(p1, p2)
f2(d)
Это возможно?
В конце концов, подумал об этом. Это просто, мне просто не хватало оператора **, чтобы распаковать словарь
Итак, мой пример:
d = dict(p1=1, p2=2)
def f2(p1,p2):
print p1, p2
f2(**d)
In[1]: def myfunc(a=1, b=2):
In[2]: print(a, b)
In[3]: mydict = {'a': 100, 'b': 200}
In[4]: myfunc(**mydict)
100 200
Несколько дополнительных деталей, которые могут быть полезны (вопросы, которые у меня возникли после прочтения и проверки):
Примеры:
Номер 1: функция может иметь параметры, которые не включены в словарь
In[5]: mydict = {'a': 100}
In[6]: myfunc(**mydict)
100 2
Номер 2: Вы не можете переопределить параметр, который уже находится в словаре
In[7]: mydict = {'a': 100, 'b': 200}
In[8]: myfunc(a=3, **mydict)
TypeError: myfunc() got multiple values for keyword argument 'a'
Номер 3: в словаре не может быть параметров, которых нет в функции.
In[9]: mydict = {'a': 100, 'b': 200, 'c': 300}
In[10]: myfunc(**mydict)
TypeError: myfunc() got an unexpected keyword argument 'c'
Как и было запрошено в комментариях, решением для номера 3 является фильтрация словаря на основе аргументов ключевых слов, доступных в функции:
In[11]: import inspect
In[12]: mydict = {'a': 100, 'b': 200, 'c': 300}
In[13]: filtered_mydict = {k: v for k, v in mydict.items() if k in [p.name for p in inspect.signature(myfunc).parameters.values()]}
In[14]: myfunc(**filtered_mydict)
100 200
Другой вариант - принять (и игнорировать) дополнительные kwargs в вашей функции:
In[15]: def myfunc2(a=None, **kwargs):
In[16]: print(a)
In[17]: mydict = {'a': 100, 'b': 200, 'c': 300}
In[18]: myfunc2(**mydict)
100
Заметьте далее, что вы можете использовать позиционные аргументы и списки практически так же, как kwargs, здесь более сложный пример, включающий все.
In[19]: def myfunc3(a, *posargs, b=2, **kwargs):
In[20]: print(a, b)
In[21]: mylist = [10, 20, 30]
In[22]: mydict = {'b': 200, 'c': 300}
In[23]: myfunc3(*mylist, **mydict)
В python это называется "распаковка", и вы можете найти немного об этом в tutorial. Документация на это отстой, я согласен, особенно из-за того, насколько это фантастически полезно.
Здесь ya go - работает только с любым другим iterable:
d = {'param' : 'test'}
def f(dictionary):
for key in dictionary:
print key
f(d)