首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往

如何在python docstring中指定不同的返回类型

在Python中,可以使用类型提示来指定函数的返回类型。类型提示是一种静态类型检查工具,可以在代码中指定变量、函数参数和返回值的类型。在函数的docstring中,可以使用类型提示来指定不同的返回类型。

以下是在Python docstring中指定不同返回类型的示例:

代码语言:python
代码运行次数:0
复制
def add(a: int, b: int) -> int:
    """
    This function takes two integers as input and returns their sum.

    Parameters:
        a (int): The first integer.
        b (int): The second integer.

    Returns:
        int: The sum of the two integers.
    """
    return a + b

在上面的示例中,函数add接受两个整数作为输入,并返回它们的和。在docstring中,使用类型提示来指定参数ab的类型为int,并指定返回值的类型也为int

对于不同的返回类型,可以根据实际情况进行指定。例如,如果函数可能返回多种类型的值,可以使用Union来指定多个类型:

代码语言:python
代码运行次数:0
复制
from typing import Union

def divide(a: int, b: int) -> Union[int, float]:
    """
    This function takes two integers as input and returns their division.
    If the division is exact, it returns an integer; otherwise, it returns a float.

    Parameters:
        a (int): The dividend.
        b (int): The divisor.

    Returns:
        Union[int, float]: The result of the division.
    """
    if a % b == 0:
        return a // b
    else:
        return a / b

在上面的示例中,函数divide接受两个整数作为输入,并返回它们的除法结果。如果除法结果是整数,则返回int类型;否则,返回float类型。在docstring中,使用Union[int, float]来指定返回类型可以是intfloat

页面内容是否对你有帮助?
有帮助
没帮助

相关·内容

领券