Príkaz return v Pythone

Majme funkciu, ktorá vypíše na obrazovku štvorec odovzdaného čísla:

def func(num):
	print(num ** 2)

V prípade, že nie je potrebné vypísať hodnotu okamžite na obrazovku, je možné ju najskôr zapísať do nejakej premennej:

res = func(3) # v premennej res je teraz 9

Na to existuje v Pythone špeciálny príkaz return, ktorý umožňuje určiť hodnotu, ktorú funkcia vracia. Pod slovom vracia sa rozumie tá hodnota, ktorá sa zapíše do premennej, ak sa jej priradí zavolaná funkcia.

Prepíšme našu funkciu tak, aby nevypisovala výsledok do konzoly, ale vracala ho do premennej:

def func(num):
	return num ** 2

Teraz zapíšme výsledok práce funkcie do premennej:

res = func(3)

Potom, ako sú dáta zapísané do premennej, je ich možné napríklad vypísať na obrazovku:

res = func(3)
print(res) # vypíše 9

Alebo je možné najskôr tieto dáta zmeniť a až potom ich vypísať na obrazovku:

res = func(3)
res = res + 1

print(res) # vypíše 10

Je možné okamžite vykonávať nejaké akcie s výsledkom práce funkcie pred zápisom do premennej:

res = func(3) + 1
print(res) # vypíše 10

Môžeme zavolať našu funkciu niekoľkokrát pre rôzne čísla a sčítať jej výsledky:

res = func(2) + func(3)
print(res) # vypíše 13

Pre skrátenie napísaného kódu je možné okamžite zavolať funkciu, odovzdaním jej do parametra print:

print(func(3))

Vytvorte funkciu, ktorá ako parameter prijme číslo a vráti kocku tohto čísla. Pomocou tejto funkcie nájdite kocku čísla 3 a zapíšte ju do premennej res.

Pomocou vytvorenej funkcie nájdite súčet kociek čísla 2 a čísla 3 a zapíšte ho do premennej res.