Escrevendo suas próprias funções

Uma função que termina sem return devolve None, e por isso print(print("hi")) mostra hi e depois None.

Duas funções que parecem iguais na tela

A maioria dos iniciantes acredita que mostrar um valor e devolvê-lo são a mesma coisa. Aqui estão duas funções que parecem "dar" a área de um retângulo. def inicia a definição de uma função: um nome, as entradas entre parênteses, dois-pontos e um corpo recuado. Execute o programa e as duas linhas de saída parecem iguais.

def show_area(w, h):
    print(w * h)

def area(w, h):
    return w * h

show_area(4, 3)
print(area(4, 3))

12
12

Agora tente usar os resultados num cálculo. A diferença aparece na hora.

def show_area(w, h):
    print(w * h)

def area(w, h):
    return w * h

a = show_area(4, 3)
b = area(4, 3)
print(a, b)
print(b * 2)

12
None 12
24

show_area imprime 12 na tela e termina. Não devolve nada, e uma função que não devolve nada retorna o valor especial None. Então a é None. area não imprime nada; ela devolve 12 à linha que a chamou, e esse 12 pode ser guardado, impresso ou dobrado. Imprimir é para as pessoas que olham a tela. Devolver é para o resto do programa. Se você quer usar um resultado depois, devolva-o.

Parâmetros são os nomes, argumentos são os valores

Em def area(w, h):, w e h são parâmetros: nomes que existem só dentro da função. Em area(4, 3), 4 e 3 são argumentos: os valores que preenchem esses nomes naquela chamada. Eles são associados por posição, então o primeiro argumento vai para o primeiro parâmetro. Chame a função com o número errado de argumentos e o Python para com um TypeError que cita o parâmetro que falta.

Um parâmetro pode ter um valor padrão, escrito com = na definição. Quem chama pode omitir esse argumento, e o padrão é usado. O tutorial do Python mostra valores padrão na seção sobre definir funções, e eles estão por toda parte nas funções embutidas. O próprio print tem um: o parâmetro end tem como padrão uma nova linha.

def greet(name, greeting="Hello"):
    return f"{greeting}, {name}!"

print(greet("Mina"))
print(greet("Jun", "Hi"))
print("no new line", end=" ")
print("here")

Hello, Mina!
Hi, Jun!
no new line here

return encerra a função na hora

Uma função para no primeiro return que alcança, mesmo que haja mais linhas abaixo. Isso faz dos retornos antecipados um jeito limpo de tratar casos especiais. A função abaixo usa a regra do ingresso da unidade sobre if, agora empacotada para que qualquer parte do programa possa pedir um preço.

def ticket_price(age):
    if age < 7:
        return 0
    if age <= 18:
        return 3000
    return 5000

family = [40, 15, 5]
total = 0
for age in family:
    total += ticket_price(age)
print(total)

8000

As variáveis criadas dentro de uma função são locais: existem enquanto a função roda e desaparecem quando ela termina. Uma variável chamada total dentro de uma função não tem nada a ver com um total fora dela. Isso permite reutilizar uma função sem se preocupar com quais nomes o resto do programa usa.