基本的Python元编程:填充字典?

时间:2012-10-29 00:37:40

标签: python exception-handling dictionary metaprogramming conditional

我一直在学习Ruby中的元编程,并且发现它非常有用。我确信我可以在Python中做同样的事情。

例如:如何使用元编程以简洁和通用的方式重写此函数?

def foo(bar=None, baz=None, qux=None, haz=None):
    txt = {}

    if bar:
        txt.update({'bar': bar})
    if baz:
        txt.update({'baz': baz})
    if qux:
        txt.update({'qux': qux})
    if haz:
        txt.update({'haz': haz})

    return txt

(这显然过于简化,实际上可能会根据各个键的值设置执行不同的任务)

2 个答案:

答案 0 :(得分:6)

您可以使用**kwargs语法:

def foo(**kwargs):
    txt = {}
    for k, v in kwargs.items():
        if v:
            txt[k] = v
    return txt

或者,如果你的墨水不足:

def foo(**kwargs):
    return dict(((k,v) for k, v in kwargs.items() if v))

正如Clement Bellot所建议的那样:

如果您想确保这些参数位于预期的设置中,则可以使用ALLOWED_PARAMS = ('bar','baz','qux','haz')常量并将if v替换为if v and k in ALLOWED_PARAMS

根据DSM的建议:

如果您运行的是Python 2.7+,则可以使用dict comprehension 此版本还提供了“允许变量”问题的替代方法:

def foo(bar=None, baz=None, qux=None, haz=None): 
    return {k:v for k,v in locals().items() if v}

假设您实际上想根据那些参数调用函数,您可以这样做:

KWARGS_TO_FUNCTIONS = {
    'bar':fn_bar,
    'baz':fn_baz, 
    'qux':fn_qux,
    'haz':fn_haz,
}

def foo(**kwargs):
    for k, v in kwargs.items():
        if v:
            try:
                KWARGS_TO_FUNCTIONS[k](v)
            except KeyError:
                pass # We didn't want that kwarg.

现在,总而言之,我并不完全确定这是'元'。

答案 1 :(得分:3)

Thomas使用**kwargs进行了一些实现。 使用if时,我认为您不需要**kwargs。在原始代码中,必须测试参数是否被省略。使用**kwargs时,字典将包含确切的给定参数,因此我们只需要

def foo(**kwargs):
    return kwargs

甚至不需要此实现 - 您可以改为使用dict()构造函数:

>>> dict(bar=1, baz=2, qux=3, haz=4)
{'qux': 3, 'haz': 4, 'baz': 2, 'bar': 1}