将Python开关重写为更紧凑的方式[重复]

Question

可能重复：
替换python中的switch语句？

假设我在Python中有一个列表：

列表=（'ADD'、'SUB'、'PUSH'、'POP'）

我想根据输入运行一个函数，该输入可以是列表中的任何值。

没有为list中的每个元素编写一个开关用例语句，有没有更紧凑的编写方法？

我的理由是，该名单今后会不断增加。

Answer 1

您也可以使用这样的模式（因为匆忙，所以无法清理）：

>>> class Test(object):
...     def test_FOO(self):
...             print 'foo'
...     
...     def test_BAR(self):
...             print 'bar'
... 
>>> def run_on(cls, name):
...     getattr(cls, 'test_%s' % name)()
... 
>>> run_on(Test(), 'FOO')
foo

Answer 2

嗯，在Python中没有开关/案例语句。

对于一个小的list，您需要使用if/elif：

def do_stuff(x, *args):
    if x == 'ADD':
        return do_add(*args)
    elif x == 'SUB':
        return do_sub(*args)
    # …
    else:
        raise RuntimeError('Never heard of {}'.format(x))

对于更大的列表，您需要确保每个案例都是一个函数（我在上面已经假设过了，但是如果您有类似return args[0]args[1]的代码，则必须将其更改为do\u add函数），并创建一个dict将名称映射到函数：

func_map = {'ADD': do_add, 'SUB': do_sub, … }

def do_stuff(x, *args):
    try:
        return func_map[x](*args)
    except KeyError:
        raise RuntimeError('Never heard of {}'.format(x))

这是因为在Python中，函数是可以像其他对象一样传递的普通对象。因此，您可以将它们存储在dict中，从dict中检索它们，然后仍然调用它们。

顺便说一句，FAQ中已经解释了这些，还有一些额外的幻想。

如果您想调用某个默认函数而不是引发错误，那么很明显如何使用If/elif/else链来实现这一点，但是如何使用dict映射呢？您可以通过将默认函数放入except块来实现，但有一种更简单的方法：只需使用dict.get方法：

def do_stuff(x, *args):
    return func_map.get(x, do_default)(*args)