Python: функции

Объявление и возврат#

Функция начинается со слова def, имени, скобок со списком параметров и
двоеточия. Тело — с отступом в четыре пробела.

def greet(name):
    """Вернуть приветствие для имени."""
    return f"Привет, {name}!"

print(greet("Аня"))     # Привет, Аня!

Строка в тройных кавычках сразу под def — документация функции, она доступна
через help(greet) и greet.__doc__. Писать её полезно даже для себя: через
месяц это единственный способ вспомнить, что функция обещает.

Если return не выполнился, функция возвращает None — ничего не «падает»,
просто результат пустой:

def add(a, b):
    print(a + b)        # печатает, но не возвращает

result = add(2, 3)      # 5
print(result)           # None

Эта путаница print и return — ошибка номер один у начинающих. print
показывает значение человеку, return отдаёт его коду.

Аргументы: позиционные, именованные, со значением по умолчанию#

def power(base, exp=2):
    return base ** exp

print(power(3))              # 9
print(power(3, 3))           # 27
print(power(exp=3, base=2))  # 8

Позиционные аргументы сопоставляются по порядку, именованные — по имени и
могут идти в любом порядке. Параметры со значением по умолчанию объявляются
только после обычных: def f(a=1, b) — синтаксическая ошибка.

Звёздочка в списке параметров делает всё, что за ней, «только по имени» — это
спасает от нечитаемых вызовов вроде save(data, True, False):

def save(data, *, overwrite=False):
    return f"{data} / перезапись: {overwrite}"

print(save("отчёт", overwrite=True))   # отчёт / перезапись: True

Переменное число аргументов#

*args собирает лишние позиционные аргументы в кортеж, **kwargs — именованные
в словарь.

def total(*args, **kwargs):
    print(args, kwargs)
    return sum(args)

print(total(1, 2, 3, скидка=5))
  # (1, 2, 3) {'скидка': 5}
  # 6

Та же звёздочка работает и при вызове — она раскрывает готовую
последовательность или словарь в аргументы:

nums = [2, 3]
print(power(*nums))                      # 8
print(power(**{"base": 5, "exp": 3}))    # 125

Изменяемое значение по умолчанию#

Выражение в умолчании вычисляется один раз — в момент объявления функции, а не
при каждом вызове. Поэтому список или словарь в умолчании живёт между вызовами:

def collect(item, box=[]):
    box.append(item)
    return box

print(collect("a"))   # ['a']
print(collect("b"))   # ['a', 'b']  — не то, чего ждали

Лечится одинаково всегда: умолчанием ставят None, а настоящее значение
создают внутри.

def collect(item, box=None):
    if box is None:
        box = []
    box.append(item)
    return box

print(collect("a"))   # ['a']
print(collect("b"))   # ['b']

Область видимости#

Имя, которому присвоили внутри функции, считается локальным для всей функции —
даже если присваивание стоит в последней строке.

counter = 0

def bump():
    global counter
    counter += 1

bump()
print(counter)      # 1

Без строки global counter тот же код упал бы с UnboundLocalError. Но
global — плохая привычка: чаще нужно принять значение аргументом и вернуть
новое. Читать глобальные значения внутри функции можно и без объявления.

Небольшая практика: напиши функцию stats(*numbers), которая возвращает
кортеж из минимума, максимума и среднего, и проверь, что она работает и от
stats(1, 2, 3), и от stats(*[1, 2, 3]). Дальше пригодятся
декораторы — они как раз надстройки над функциями,
и исключения, чтобы аккуратно сообщать об
ошибочных аргументах.

План по этапам

1. Простая функция с returnНаписать функцию, которая принимает два числа и возвращает их сумму; вывести результат через print.
2. Умолчания и именованные аргументыДобавить параметр со значением по умолчанию и вызвать функцию тремя способами.
3. Собрать *argsФункция с переменным числом аргументов, которая считает среднее; вызвать её с раскрытием списка.
4. Воспроизвести ловушкуВыполнить пример с box=[] дважды подряд и убедиться, что список накапливает значения.
5. Переписать безопасноЗаменить умолчание на None и проверить, что вызовы больше не влияют друг на друга.

Проверь себя

Источники

• Учебник Python: определение функцийРаздел про def, умолчания, *args и **kwargs
  бесплатно
• Справочник встроенных функцийsum, min, max, sorted и другие функции для проверки своих решений
  бесплатно
• PEP 8 — стиль кодаКак называть функции и параметры
  бесплатно

Было полезно?

Да Нет