如何使用callables处理python字典?

时间:2013-01-30 17:43:18

标签: python dictionary

我想创建一个python目录,其值将单独计算。

因此,例如,在以下非工作示例中我定义了

a = {'key1': 'value1', 'key2': 42, 'key3': foo(20)}

,例如

def foo(max):
    """Returns random float between 0 and max."""
    return max*random.random()

并且在处理dict之后我想要例如。

a_processes = {'key1': 'value1', 'key2': 42, 'key3': 12.238746374}

该示例不起作用,因为密钥key3的值立即被评估,foo(20)不可调用。它的工作方式是使用像

这样的东西
a = {'key1': 'value1', 'key2': 42, 'key3': foo}

但是这里foo会错过它的论点。处理此问题的一种方法是按以下方式定义dict

a = {'key1': 'value1', 'key2': 42, 'key3': [foo, 20]}

使用以下处理方案

a_processed = dict([k,process(v)] for k,v in a.items())

其中process是一个有意义的函数,检查它的参数是否是一个列表,或者第一个元素是一个可调用的函数,用剩余的参数调用它。

我的问题是关于实现我的基本想法的更好方式/想法?

2 个答案:

答案 0 :(得分:8)

使用functools.partial()创建一个callable,将一组参数应用于另一个callable:

from functools import partial

a = {'key1': 'value1', 'key2': 42, 'key3': partial(foo, 20)}

现在key3值是一个可调用对象; a['key3']()将依次调用foo(20)并返回结果。

您仍然需要处理'这本词典(使用callable() function进行测试:

{k: v() if callable(v) else v for k, v in a.iteritems()}

当然。如果您使用的是Python 3,请改用a.items()

functools.partial设置允许您根据需要添加其他参数(前提是foo()支持):

a['key3']('additional', arguments='accepted')

您可以甚至创建一个字典子类,在访问密钥时自动执行此操作:

def CallablesDict(dict):
    def __getitem__(self, key):
        value = super(CallablesDict, self).__getitem__(key)
        return value() if callable(value) else value

然后将a定义为:

a = CallablesDict({'key1': 'value1', 'key2': 42, 'key3': partial(foo, 20)})

print a['key3']  # calls foo(20)

答案 1 :(得分:5)

我经常在这里使用无参数lambda,因为我喜欢语法:

>>> def foo(x):
...     return x*100
... 
>>> a = {'key1': 'value1', 'key2': 42, 'key3': lambda: foo(20)}
>>> a
{'key3': <function <lambda> at 0x96ae80c>, 'key2': 42, 'key1': 'value1'}
>>> {k: v() if callable(v) else v for k,v in a.iteritems()}
{'key3': 2000, 'key2': 42, 'key1': 'value1'}