python：通过if/elif语句传递值[duplicate]

Question

我有一个函数可以返回三个值中的一个：a、b或c。

def f():
    return x # a, b, or c

我需要根据f（）：A、B或C的返回值执行不同的语句。据我所知，最好的方法是：

v = f()
if v == a:
    # Do A
elif v == b:
    # Do B
else:
    # Do C

如果不引入这个新变量v，是否有其他方法来计算f（）的返回值<也许是这样的吧？

if f() == a:
    # Do A
elif _value_from_above == b:
    # Do B
else:
    # Do C

Answer 1

如果/elif和Python没有开关语句，就没有办法做到这一点。你需要临时变量。

如果您知道a、b和c的确切值，也可以使用分派表。它更详细一点，但是如果您有很多可能的分支，它将更快，并且可能更容易遵循。

要制作调度表：

• 为每个分支创建一个函数
• 制作一本字典，将a、b、c映射到它们相应的函数
• 将所有代码替换为return DISPATCH_TABLE[f（）]（）

你可以在这里找到一些例子。

Answer 2

如果你只想映射答案（不需要进一步计算），那么你可以用一个简单的映射函数来解决它（类似于Junichiro的答案）：

import random

def f():
    foo = ['a', 'b', 'c', 'd', 'e']
    return random.choice(foo)

if __name__ == '__main__':
    from_to = dict(a=2, b=1, c=3, d=4, e=5)
    print(from_to.get(f(), None))

如果您在f（）和之间有一个不可预见的值。get（f（），None）将默认为None

Answer 3

可能不是您想要的，但一种选择是使用函数字典：

def do_a();
    ..
def do_b();
    ..
def do_c();
    ..

strategy = {
    a: do_a,
    b: do_b
    c: do_c
}

strategy[f()]()

但是通常情况下，您需要一些错误处理，以防找不到字典条目，在这种情况下，您可能最终会有一个单独的变量，而且您现在必须将每个案例提取到自己的函数中。

如果只有两种情况，您可以使用三元运算符：

do_a() if f() == a else do_b()

虽然你可能已经知道了这一点，加上你的问题有两个以上的情况，加上很多人不喜欢三元运算符，但是不管怎样。

如果您有一种带有switch语句的语言，比如C#/Java/C等，或者一种模式匹配语句，比如Haskell/OCaml/F#/Scala/Erlang等，那么您可以使用它们。模式匹配语句如下所示：

match f() with
| a ->
    // do something
    // and maybe something else too
| b ->
    // do some other thing
    // and yet some other thing

但是Python中没有这样的特性。