Ersetzungen für switch-Anweisung in Python?

Question

Ich möchte eine Funktion in Python schreiben, die je nach dem Wert eines Eingabeindexes verschiedene feste Werte zurückgibt.

In anderen Sprachen würde ich ein switch o case Anweisung, aber Python scheint nicht über eine switch Aussage. Was sind die empfohlenen Python-Lösungen in diesem Szenario?

Answer 1

Ich habe eine Schaltkasten Implementierung, die nicht ganz mit ifs extern (es verwendet immer noch eine wenn in der Klasse).

class SwitchCase(object):
    def __init__(self):
        self._cases = dict()

    def add_case(self,value, fn):
        self._cases[value] = fn

    def add_default_case(self,fn):
        self._cases['default']  = fn

    def switch_case(self,value):
        if value in self._cases.keys():
            return self._cases[value](value)
        else:
            return self._cases['default'](0)

Verwenden Sie es so:

from switch_case import SwitchCase
switcher = SwitchCase()
switcher.add_case(1, lambda x:x+1)
switcher.add_case(2, lambda x:x+3)
switcher.add_default_case(lambda _:[1,2,3,4,5])

print switcher.switch_case(1) #2
print switcher.switch_case(2) #5
print switcher.switch_case(123) #[1, 2, 3, 4, 5]

Answer 2

Wenn Sie wirklich nur einen vorgegebenen, festen Wert zurückgeben wollen, könnten Sie ein Wörterbuch mit allen möglichen Eingabeindizes als Schlüssel und den entsprechenden Werten erstellen. Außerdem möchten Sie vielleicht nicht wirklich, dass eine Funktion dies tut - es sei denn, Sie berechnen den Rückgabewert irgendwie.

Oh, und wenn Sie Lust haben, etwas wie einen Schalter zu benutzen, sehen Sie sich aquí .

Answer 3

Ähnlich wie diese Antwort von abarnert Hier ist eine Lösung speziell für den Anwendungsfall des Aufrufs einer einzigen Funktion für jeden "Fall" in der Weiche, unter Vermeidung der lambda o partial für Ultra-Konzisität und kann gleichzeitig mit Schlüsselwortargumenten umgehen:

class switch(object):
    NO_DEFAULT = object()

    def __init__(self, value, default=NO_DEFAULT):
        self._value = value
        self._result = default

    def __call__(self, option, func, *args, **kwargs):
        if self._value == option:
            self._result = func(*args, **kwargs)
        return self

    def pick(self):
        if self._result is switch.NO_DEFAULT:
            raise ValueError(self._value)

        return self._result

Beispiel für die Verwendung:

def add(a, b):
    return a + b

def double(x):
    return 2 * x

def foo(**kwargs):
    return kwargs

result = (
    switch(3)
    (1, add, 7, 9)
    (2, double, 5)
    (3, foo, bar=0, spam=8)
    (4, lambda: double(1 / 0))  # if evaluating arguments is not safe
).pick()

print(result)

Beachten Sie, dass dies eine Verkettung von Aufrufen ist, d.h. switch(3)(...)(...)(...) . Setzen Sie keine Kommas dazwischen. Es ist auch wichtig, alles in einen Ausdruck zu packen, weshalb ich zusätzliche Klammern um den Hauptaufruf für die implizite Zeilenfortsetzung gesetzt habe.

Das obige Beispiel löst einen Fehler aus, wenn Sie einen Wert einschalten, der nicht behandelt wird, z. B. switch(5)(1, ...)(2, ...)(3, ...) . Sie können stattdessen einen Standardwert angeben, z. B. switch(5, default=-1)... gibt zurück. -1 .

Answer 4

Obwohl es bereits genügend Antworten gibt, möchte ich auf eine einfachere und leistungsfähigere Lösung hinweisen:

class Switch:
    def __init__(self, switches):
        self.switches = switches
        self.between = len(switches[0]) == 3

    def __call__(self, x):
        for line in self.switches:
            if self.between:
                if line[0] <= x < line[1]:
                    return line[2]
            else:
                if line[0] == x:
                    return line[1]
        return None

if __name__ == '__main__':
    between_table = [
        (1, 4, 'between 1 and 4'),
        (4, 8, 'between 4 and 8')
    ]

    switch_between = Switch(between_table)

    print('Switch Between:')
    for i in range(0, 10):
        if switch_between(i):
            print('{} is {}'.format(i, switch_between(i)))
        else:
            print('No match for {}'.format(i))

    equals_table = [
        (1, 'One'),
        (2, 'Two'),
        (4, 'Four'),
        (5, 'Five'),
        (7, 'Seven'),
        (8, 'Eight')
    ]
    print('Switch Equals:')
    switch_equals = Switch(equals_table)
    for i in range(0, 10):
        if switch_equals(i):
            print('{} is {}'.format(i, switch_equals(i)))
        else:
            print('No match for {}'.format(i))

Ausgabe:

Switch Between:
No match for 0
1 is between 1 and 4
2 is between 1 and 4
3 is between 1 and 4
4 is between 4 and 8
5 is between 4 and 8
6 is between 4 and 8
7 is between 4 and 8
No match for 8
No match for 9

Switch Equals:
No match for 0
1 is One
2 is Two
No match for 3
4 is Four
5 is Five
No match for 6
7 is Seven
8 is Eight
No match for 9

Answer 5

Und eine weitere Möglichkeit:

def fnc_MonthSwitch(int_Month): #### Define a function take in the month variable 
    str_Return ="Not Found"     #### Set Default Value 
    if int_Month==1:       str_Return = "Jan"   
    if int_Month==2:       str_Return = "Feb"   
    if int_Month==3:       str_Return = "Mar"   
    return str_Return;          #### Return the month found  
print ("Month Test 3:  " + fnc_MonthSwitch( 3) )
print ("Month Test 14: " + fnc_MonthSwitch(14) )