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

Die unten stehende Originalantwort wurde 2008 verfasst. Seitdem wurde mit Python 3.10 (2021) die match - case Anweisung, die eine erstklassige Implementierung eines "Schalters" für Python bietet. Zum Beispiel:

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

El match - case Anweisung ist wesentlich leistungsfähiger als dieses einfache Beispiel.

---

Sie könnten ein Wörterbuch benutzen:

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

Answer 2

Wenn Sie Standardwerte wünschen, können Sie das Wörterbuch get(key[, default]) Funktion:

def f(x):
    return {
        'a': 1,
        'b': 2
    }.get(x, 9)    # 9 will be returned default if x is not found

Answer 3

Ich habe es immer gerne so gemacht

result = {
  'a': lambda x: x * 5,
  'b': lambda x: x + 7,
  'c': lambda x: x - 2
}[value](x)

Von hier aus

Answer 4

Zusätzlich zu den Wörterbuchmethoden (die ich übrigens sehr mag), können Sie auch if - elif - else um die switch / case / default Funktionalität:

if x == 'a':
    # Do the thing
elif x == 'b':
    # Do the other thing
if x in 'bc':
    # Fall-through by not using elif, but now the default case includes case 'a'!
elif x in 'xyz':
    # Do yet another thing
else:
    # Do the default

Dies ist natürlich nicht identisch mit Schalter/Gehäuse - man kann nicht so einfach durchfallen, wie wenn man die break Anweisung, aber Sie können auch einen komplizierteren Test durchführen. Seine Formatierung ist schöner als eine Reihe von verschachtelten if s, auch wenn es funktionell gesehen eher dem entspricht.

Answer 5

Python >= 3.10

Wow, Python 3.10+ hat jetzt eine match / case Syntax, die wie folgt lautet switch/case und mehr!

PEP 634 -- Struktureller Musterabgleich

Ausgewählte Merkmale von match/case

1 - Werte abgleichen:

Der Abgleich von Werten ist ähnlich wie eine einfache switch/case in einer anderen Sprache:

match something:
    case 1 | 2 | 3:
        # Match 1-3.
    case _:
        # Anything else.
        # 
        # Match will throw an error if this is omitted 
        # and it doesn't match any of the other patterns.

2 - Strukturelle Muster anpassen:

match something:
    case str() | bytes():  
        # Match a string like object.
    case [str(), int()]:
        # Match a `str` and an `int` sequence 
        # (`list` or a `tuple` but not a `set` or an iterator). 
    case [_, _]:
        # Match a sequence of 2 variables.
        # To prevent a common mistake, sequence patterns don’t match strings.
    case {"bandwidth": 100, "latency": 300}:
        # Match this dict. Extra keys are ignored.

3 - Erfassen von Variablen

Parsen eines Objekts; Speichern als Variablen:

match something:
    case [name, count]
        # Match a sequence of any two objects and parse them into the two variables.
    case [x, y, *rest]:
        # Match a sequence of two or more objects, 
        # binding object #3 and on into the rest variable.
    case bytes() | str() as text:
        # Match any string like object and save it to the text variable.

Capture-Variablen können beim Parsen von Daten (z. B. JSON oder HTML) nützlich sein, die in einem von vielen verschiedenen Mustern vorliegen können.

Das Erfassen von Variablen ist eine Funktion. Aber es bedeutet auch, dass Sie gepunktete Konstanten verwenden müssen (Bsp: COLOR.RED ) nur. Andernfalls wird die Konstante wie eine Capture-Variable behandelt und überschrieben.

Weitere Verwendungsbeispiele :

match something:
    case 0 | 1 | 2:
        # Matches 0, 1 or 2 (value).
        print("Small number")
    case [] | [_]:
        # Matches an empty or single value sequence (structure).
        # Matches lists and tuples but not sets.
        print("A short sequence")
    case str() | bytes():
        # Something of `str` or `bytes` type (data type).
        print("Something string-like")
    case _:
        # Anything not matched by the above.
        print("Something else")

---

Python <= 3.9

Mein bevorzugtes Python-Rezept für Switch/Case war:

choices = {'a': 1, 'b': 2}
result = choices.get(key, 'default')

Kurz und einfach für einfache Szenarien.

Im Vergleich zu 11+ Zeilen C-Code:

// C Language version of a simple 'switch/case'.
switch( key ) 
{
    case 'a' :
        result = 1;
        break;
    case 'b' :
        result = 2;
        break;
    default :
        result = -1;
}

Sie können sogar mehrere Variablen zuweisen, indem Sie Tupel verwenden:

choices = {'a': (1, 2, 3), 'b': (4, 5, 6)}
(result1, result2, result3) = choices.get(key, ('default1', 'default2', 'default3'))