Курс Python → Типы возвращаемых значений в Python

Для работы с функциями, которые могут возвращать значения различных типов, в Python можно использовать типы typing.Optional или typing.Union. Например, если у вас есть переменная, которая может быть либо строкой, либо None, то вы можете указать тип Optional[str]. Это позволит явно указать, что функция может вернуть либо строку, либо None.

Для случаев, когда функция может возвращать значения нескольких различных типов, можно воспользоваться типом Union. Например, если функция может вернуть как строку, так и целое число, то вы можете указать тип Union[str, int]. Таким образом, вы укажете, что функция может вернуть либо строку, либо целое число.

Использование типов возвращаемых значений позволяет улучшить читаемость кода и облегчить его понимание. Другие разработчики, работающие с вашим кодом, смогут быстро понять, какие типы данных могут быть возвращены функцией, что ускорит процесс разработки и отладки.

from typing import Optional, Union

def example_function(input: str) -> Optional[str]:
    if input == "example":
        return "This is an example"
    else:
        return None

def another_function(input: int) -> Union[str, int]:
    if input % 2 == 0:
        return "Even number"
    else:
        return input

В приведенном примере функция example_function принимает строку и возвращает либо строку, либо None, в зависимости от значения входного аргумента. Функция another_function принимает целое число и возвращает либо строку, либо само число, в зависимости от условия внутри функции.

Твои коллеги будут рады, поделись в