Python中switch语句的替换？

我想用Python编写一个函数，根据输入索引的值返回不同的固定值。

在其他语言中，我会使用switch或case语句，但Python似乎没有switch语句。在此场景中，推荐的Python解决方案是什么？

当前回答

如果要搜索额外的语句，如“switch”，我构建了一个扩展Python的Python模块。它被称为ESPY“增强的Python结构”，可用于Python2.x和Python3.x。

例如，在这种情况下，switch语句可以由以下代码执行：

macro switch(arg1):
    while True:
        cont=False
        val=%arg1%
        socket case(arg2):
            if val==%arg2% or cont:
                cont=True
                socket
        socket else:
            socket
        break

可以这样使用：

a=3
switch(a):
    case(0):
        print("Zero")
    case(1):
        print("Smaller than 2"):
        break
    else:
        print ("greater than 1")

所以espy在Python中将其翻译为：

a=3
while True:
    cont=False
    if a==0 or cont:
        cont=True
        print ("Zero")
    if a==1 or cont:
        cont=True
        print ("Smaller than 2")
        break
    print ("greater than 1")
    break

2010-12-06 15:08:12

其他回答

Python 3.10（2021）引入了match-case语句，该语句提供了Pythons“switch”的一流实现。例如：

def f(x):
    match x:
        case 'a':
            return 1
        case 'b':
            return 2
        case _:
            return 0   # 0 is the default case if x is not found

match-case语句比这个简单的示例强大得多。

以下原始答案写于2008年，当时还未提供匹配案例：

你可以用字典：

def f(x):
    return {
        'a': 1,
        'b': 2,
    }[x]

2008-09-13 00:38:24

还可以使用列表存储案例，并通过select调用相应的函数-

cases = ['zero()', 'one()', 'two()', 'three()']

def zero():
  print "method for 0 called..."
def one():
  print "method for 1 called..."
def two():
  print "method for 2 called..."
def three():
  print "method for 3 called..."

i = int(raw_input("Enter choice between 0-3 "))

if(i<=len(cases)):
  exec(cases[i])
else:
  print "wrong choice"

也在螺丝台上进行了解释。

2013-06-06 14:01:46

扩展Greg Hewgill的答案-我们可以使用装饰器封装字典解决方案：

def case(callable):
    """switch-case decorator"""
    class case_class(object):
        def __init__(self, *args, **kwargs):
            self.args = args
            self.kwargs = kwargs

        def do_call(self):
            return callable(*self.args, **self.kwargs)

return case_class

def switch(key, cases, default=None):
    """switch-statement"""
    ret = None
    try:
        ret = case[key].do_call()
    except KeyError:
        if default:
            ret = default.do_call()
    finally:
        return ret

然后可以将其与@case decorator一起使用

@case
def case_1(arg1):
    print 'case_1: ', arg1

@case
def case_2(arg1, arg2):
    print 'case_2'
    return arg1, arg2

@case
def default_case(arg1, arg2, arg3):
    print 'default_case: ', arg1, arg2, arg3

ret = switch(somearg, {
    1: case_1('somestring'),
    2: case_2(13, 42)
}, default_case(123, 'astring', 3.14))

print ret

好消息是，这已经在NeoPySwitch模块中完成。只需使用pip进行安装：

pip install NeoPySwitch

2017-04-21 07:17:38

作为Mark Biek答案的一个小变化，对于像这样的不常见情况，用户有一堆函数调用要延迟，而参数要打包（而且不值得构建一堆不符合逻辑的函数），而不是这样：

d = {
    "a1": lambda: a(1),
    "a2": lambda: a(2),
    "b": lambda: b("foo"),
    "c": lambda: c(),
    "z": lambda: z("bar", 25),
    }
return d[string]()

…您可以这样做：

d = {
    "a1": (a, 1),
    "a2": (a, 2),
    "b": (b, "foo"),
    "c": (c,)
    "z": (z, "bar", 25),
    }
func, *args = d[string]
return func(*args)

这当然更短，但它是否更可读是一个悬而未决的问题…

我认为从lambda转换为partial可能更容易理解（虽然不是更简单）：

d = {
    "a1": partial(a, 1),
    "a2": partial(a, 2),
    "b": partial(b, "foo"),
    "c": c,
    "z": partial(z, "bar", 25),
    }
return d[string]()

…它的优点是可以很好地处理关键字参数：

d = {
    "a1": partial(a, 1),
    "a2": partial(a, 2),
    "b": partial(b, "foo"),
    "c": c,
    "k": partial(k, key=int),
    "z": partial(z, "bar", 25),
    }
return d[string]()

2018-06-04 18:41:10