Wielokrotny wybór w języku Python

Podczas tworzenia programu zdarzają się sytuacje, w których potrzebujemy zrealizować jeden z wielu scenariuszy działań.

Przeanalizujmy przykład, gdzie mamy wypisać informację, w którym z następujących przedziałów: $(10,15],(15,21],(21,50]$ zawiera się dana liczba.

Do zrealizowania takiego scenariusza możemy zastosować konstrukcję if/elif/else, ponieważ wybieramy jeden z wielu przedziałów.

Ogólną postać instrukcji znajdziesz w e‑materiale Instrukcja warunkowa w języku PythonPb35akOxSInstrukcja warunkowa w języku Python.

wartosc = 23

if 10 < wartosc and wartosc <= 15:
	print("Przedział 1")
elif wartosc > 15 and wartosc <= 21:
	print("Przedział 2")
elif wartosc > 21 and wartosc <= 50:
	print("Przedział 3")
else:
	print("Liczba nie zawiera się w żadnym z przedziałów")

wartosc = 23

if wartosc > 10 and wartosc <= 15:
    print("Przedział 1")
if wartosc > 15 and wartosc <= 21:
    print("Przedział 2")
if wartosc > 21 and wartosc <= 50:
    print("Przedział 3")
if wartosc <= 10 or wartosc > 50:
    print("Liczba nie zawiera się w żadnym z przedziałów")

def funkcja_1():
	print("Funkcja 1.")

def funkcja_2():
	print("Funkcja 2.")

def funkcja_3():
	print("Funkcja 3.")

def funkcja_x():
	print("Funkcja x.")


wybor = input("Podaj numer: ")
wynik = int(wybor)

if wynik == 1:
	funkcja_1()
elif wynik == 2:
	funkcja_2()
elif wynik == 3:
	funkcja_3()
else:
	funkcja_x()

# Efekt wykonania:

# Podaj numer: 2
# Funkcja 2.

def funkcja_1():
	print("Funkcja 1.")

def funkcja_2():
	print("Funkcja 2.")

def funkcja_3():
	print("Funkcja 3.")

def funkcja_x():
	print("Funkcja x.")


wybor = input("Podaj numer: ")
wynik = int(wybor)

# tworzymy słownik zawierający jako wartości nazwy funkcji
slownik = {1: funkcja_1, 2: funkcja_2, 3: funkcja_3}

# jeśli wybór jest kluczem słownika, wykonujemy funkcję
if wynik in slownik:
	slownik[wynik]()
# w innym przypadku wykonujemy konkretną funkcję
else:
	funkcja_x()

Trójargumentowe wyrażenie wyboru

Używane jest w sytuacjach, gdy wystarczy zwrócić prostą informację zależną od jednego warunku. W takim przypadku rozbudowywanie kodu jest niecelowe.

Oto wyrażenie trójargumentowe zapisane w języku Python:

wartość_gdy_prawda if warunek_logiczny else wartość_gdy_fałsz

Instrukcja taka odpowiada poleceniom:

if warunek_logiczny:
	wartość_gdy_prawda
else:
	wartość_gdy_fałsz

Porównajmy kody dwóch funkcji. Mają one sprawdzić, czy tekst przekazany im jako argument jest palindromem. Sprawdzenie, czy słowo jest palindromem, wykonane jest przy użyciu wyrażenia indeksującego.

Pierwsza funkcja wykorzystuje zwykłą konstrukcję if/else:

def czy_palindrom(tekst):
	if tekst == tekst[::-1]:
		return True
	else:
		return False

Druga funkcja wykorzystuje wyrażenie trójargumentowe:

def czy_palindrom(tekst):
	return True if tekst == tekst[::-1] else False

Obie powyższe funkcje mogą być zapisane również następująco:

def czy_palindrom(tekst):
	return tekst == tekst[::-1]

Innym przykładem zastosowania wyrażenia trójargumentowego jest podanie jednego z dwóch komunikatów, zależnych od spełnienia pewnego warunku. Posłużymy się funkcją print():

wartosc = 15
print('Dużo' if wartosc > 20 else 'Mało', 'pieniędzy.')
# Mało pieniędzy.

wartosc = 23
print('Dużo' if wartosc > 20 else 'Mało', 'pieniędzy.')
# Dużo pieniędzy.

Instrukcja match/case

W wersji 3.10 języka Python została wprowadzona instrukcja match/case. Instrukcja ta jako argument przyjmuje wyrażenie i porównuje jego wartość ze zdefiniowanymi schematami. Jeśli wyrażenie jest zgodne ze zdefiniowanym schematem, wykonywane są związane z nim instrukcje.

Instrukcja match/case zapisana za pomocą języka Python wygląda następująco:

match numer_ucznia:
	case 1:
		# wypisz informacje o uczniu numer 1.
	case 2:
		# wypisz informacje o uczniu numer 2.
	case 3:
		# wypisz informacje o uczniu numer 3.  
	case 4:
		# wypisz informacje o uczniu numer 4.

numerUucznia = 3

match numer_ucznia:
	case 1:
		# wypisz informacje o uczniu numer 1.
	case 2:
		# wypisz informacje o uczniu numer 2.
	case 3:
		# wypisz informacje o uczniu numer 3.  
	case 4:
		# wypisz informacje o uczniu numer 4.

Instrukcja match/case pobiera obiekt (numer_ucznia), testuje go względem jednego lub więcej wzorców dopasowania (case 1, case 2 etc.) i wykonuje akcję, jeśli znajdzie dopasowanie.

Po każdym słowie kluczowym case następuje wzorzec dopasowania.

Python wykonuje dopasowania, przechodząc przez listę przypadków od góry do dołu. Przy pierwszym dopasowaniu Python wykonuje instrukcje w odpowiadającym im bloku przypadków, a następnie przeskakuje na koniec bloku dopasowania i kontynuuje dalszą część programu. Oznacza to, że po znalezieniu pierwszego pasującego przypadku, kończy się działanie instrukcji match/case.

Załóżmy, że chcemy napisać program, który wyświetli komunikat Wakacje!, jeśli użytkownik poda numer wakacyjnego miesiąca (zakładamy, że wakacyjne miesiące to lipiec i sierpień) lub To nie wakacje :-(, jeśli poda numer jakiegokolwiek innego miesiąca.

miesiac = 2

match miesiac:
	case 1:
		print("To nie wakacje :-(")
	case 2:
		print("To nie wakacje :-(")
	case 3:
		print("To nie wakacje :-(")
	case 4:
		print("To nie wakacje :-(")
	case 5:
		print("To nie wakacje :-(")
	case 6:
		print("To nie wakacje :-(")        
	case 7:
		print("Wakacje!")
	case 8:
		print("Wakacje!")
	case 9:
		print("To nie wakacje :-(")
	case 10:
		print("To nie wakacje :-(")
	case 11:
		print("To nie wakacje :-(")
	case 12:
		print("To nie wakacje :-(")

Łączenie warunków

Używając operatora |, oznaczającego alternatywę, możemy łączyć warunki.

miesiac = 2

match miesiac:
	case 1 | 2 | 3 | 4 | 5 | 6 | 9 | 10 | 11 | 12:
		print("To nie wakacje :-(")
	case 7 | 8:
		print("Wakacje!")

Domyślny przypadek

W instrukcji match/case możemy podać przypadek _, który oznacza „w każdym innym przypadku”, czyli jeżeli wartość zmiennej podanej do sprawdzenia nie będzie równa żadnemu z ustawionych przypadków, program wykona fragment kodu zawarty w poleceniu.

Przeanalizuj przykład:

miesiac = 27

match miesiac:
	case 1 | 2 | 3 | 4 | 5 | 6 | 9 | 10 | 11 | 12:
		print("To nie wakacje :-(")
	case 7 | 8:
		print("Wakacje!")
	case _:
		print("Wybrano liczbę spoza przedziału [1, 12]")

Słownik