Аргументы, несколько результатов и lambda

Цену со скидкой считают постоянно, но скидка есть не всегда. Писать каждый раз final_price(200, 0) неудобно, а иногда функция должна вернуть сразу два ответа. У функции есть договорённость — что она принимает, что возвращает и меняет ли данные; значения по умолчанию, именованные аргументы и несколько результатов делают её удобной.

Разберём это на цене со скидкой final_price(price, discount=0), а в конце оформим функцией сквозную задачу курса.

Параметр со значением по умолчанию можно не передавать, а аргумент можно назвать по имени.

def final_price(price, discount=0):    return price * (100 - discount) / 100 print(final_price(200))print(final_price(200, 10))print(final_price(discount=10, price=200))

discount=0 в определении — значение по умолчанию. В вызове именованные аргументы уточняют, какому параметру соответствует значение. Вот как параметры получают значения в трёх вызовах:

Обычные позиционные аргументы записывают перед именованными. Одному параметру нельзя передать значение дважды:

def final_price(price, discount=0):    return price * (100 - discount) / 100 print(final_price(200, price=100))

Значения по умолчанию вычисляются при выполнении определения, а не заново для каждого вызова. Поэтому изменяемый список в параметре по умолчанию будет общим между вызовами:

def add_item_bad(item, items=[]):    items.append(item)    return items print(add_item_bad(1))print(add_item_bad(2))

Второй вызов увидел единицу из первого: список по умолчанию создан один раз. Нажми «Запуск» и пройди программу по шагам: во втором вызове items ведёт к тому же списку, что и в первом, — он создан, когда выполнилось def, и уже содержит 1.

Правильный приём — None по умолчанию и новый список внутри:

def add_item(item, items=None):    if items is None:        items = []    items.append(item)    return items print(add_item(1))print(add_item(2))

None обозначает отсутствие переданного списка. В таком случае создаётся новый. Если пользователь явно передал список, функция изменит именно его. Эта договорённость должна быть понятна из описания.

Функция может вернуть несколько значений — на самом деле один кортеж.

def quotient_and_remainder(a, b):    return a // b, a % b q, r = quotient_and_remainder(23, 5)print(q, r)

Фактически возвращается один кортеж, который распаковывается. Для этой конкретной операции уже существует divmod(a, b), но пример показывает общую механику.

def total_of(*values):    return sum(values) print(total_of(2, 3, 5))print(total_of())

Функцию можно передать как значение; lambda создаёт короткую функцию из одного выражения. Имя функции можно передать другой функции. Без скобок это сама функция; со скобками — результат её вызова.

def last_digit(number):    return abs(number) % 10 rule = last_digitprint(rule(47))short_rule = lambda number: abs(number) % 10print(short_rule(-123))

lambda number: abs(number) % 10 создаёт короткую функцию с одним выражением. Она делает то же, что last_digit, только у неё нет собственного имени. Она автоматически возвращает результат выражения. Несколько инструкций в lambda не помещают: используй def. Чаще всего lambda передают другой функции — например, как правило сортировки в уроке о сортировке.

Хорошая функция начинается с договорённости о входе, результате и особых случаях. Сначала сформулируй договорённость: что функция принимает, что возвращает, какие случаи допускаются, меняет ли аргументы. Например: «принимает целое число, возвращает сумму цифр его модуля».

def digit_sum(number):    """Вернуть сумму десятичных цифр модуля целого числа."""    number = abs(number)    total = 0    while number > 0:        total += number % 10        number //= 10    return total print(digit_sum(-5072), digit_sum(0))

Первая строка текста в теле функции — строка документации, или docstring. Её может показывать help(digit_sum). Тройные кавычки позволяют записывать многострочный текст; это не особый вид комментария.

Возврат ставят после цикла, когда просмотрены все элементы. Напишем count_even(numbers), возвращающую количество чётных элементов без изменения списка:

def count_even(numbers):    count = 0    for number in numbers:        if number % 2 == 0:            count += 1    return count print(count_even([0, 3, -4, 7]))print(count_even([]))

Возврат стоит после цикла. Если поставить его внутри тела, функция закончится после первого элемента. Для пустого списка корректно возвращается 0.

Сквозная задача курса: сумма и наибольшее — одной функцией с двумя результатами. Цикл из урока о списках оформим функцией с договорённостью «принимает непустой список чисел, возвращает сумму и наибольшее»:

def total_and_best(numbers):    total = 0    best = numbers[0]    for x in numbers:        total += x        if x > best:            best = x    return total, best total, best = total_and_best([5, -2, 10, 7])print(total, best)

return total, best возвращает один кортеж, а строка вызова раскладывает его на два имени. Слово «непустой» в договорённости важно: для пустого списка numbers[0] остановит программу с IndexError. Теперь функцию можно вызвать для любого списка — из ввода, из файла, из другой функции.