课堂内的Pythonic开关

时间:2017-10-24 10:47:28

标签: python dictionary switch-statement

我正在尝试在python中构建一个开关逻辑。

基于Replacements for switch statement in Python?我构建了以下代码:

def func(string):
    print(string)

class Switch:
    def __init__(self):
        pass

    def first_switch(self, case):
        return {
             '1': func('I am case 1'),
             '2': func('I am case 2'),
             '3': func('I am case 3'),
             }.get(case, func('default'))


switch = Switch()
switch.first_switch('2')

这将给我输出:

I am case 1
I am case 2
I am case 3
default

我预计输出为

I am case 2

这个字典大小写逻辑是否只适用于类定义之外,还是我错过了一些额外的代码?看起来在函数调用中评估所有字典键值对。

1 个答案:

答案 0 :(得分:7)

在构建字典时,您总是调用所有这些函数。它与课程无关。

d = {'foo': bar()}
此处调用

bar,并将其返回值分配给d['foo']。请记住,这不是switch声明;它是一个字典文字,用于模拟switch语句。

在您的情况下,该函数不是可变的,因此根本不必包括:

arg = {
    '1': 'I am case 1',
    '2': 'I am case 2',
    '3': 'I am case 3',
}.get(case, 'default')
func(arg)

如果函数也是可变的,你想让字典值callables在调用时调用函数

{'1': lambda: func('I am case 1'), ...}.get(...)()
                    # call the function you got ^^

或者也许:

from functools import partial

{'1': partial(func, 'I am case 1'), ...}.get(...)()