Python 设置返回类型

在Python中，函数的返回类型没有严格的限制，可以返回任何类型的对象。但是，为了提高代码的可读性和可维护性，有时需要将返回类型标注出来。从Python 3.5开始，可以使用函数注解来标注返回类型。

函数注解

函数注解是在函数定义语句上方的一行注释，用于说明函数的参数和返回值的类型。语法格式如下所示：

def funcname(param1: type1, param2: type2) -> return_type:
    function body

其中，param1: type1表示函数的第一个参数的类型为type1，param2: type2表示函数的第二个参数的类型为type2，-> return_type表示函数的返回类型为return_type。

下面是一个示例：

def add(a: int, b: int) -> int:
    return a + b

result = add(1, 2)
print(result)  # 输出 3

在上面的示例中，add函数标注了参数的类型int和返回值的类型int，如果传入的参数类型不是int，将会在运行时抛出类型错误。

类型注解

除了函数注解外，还可以使用类型注解来标注变量的类型。类型注解是在变量名后面的冒号和类型表达式，语法格式如下：

variable_name: type = value

其中，variable_name表示变量名，type表示变量的类型，value表示变量的值。示例如下：

a: int = 1
b: str = 'hello'
c: List[int] = [1, 2, 3]

print(type(a))  # 输出 <class 'int'>
print(type(b))  # 输出 <class 'str'>
print(type(c))  # 输出 <class 'list'>

在上面的示例中，a、b和c分别标注了它们的类型，如果对它们赋值时类型不符，将会在运行时抛出类型错误。

结语

虽然Python不强制限制函数和变量的类型，但是类型注解可以让代码更加易于阅读和维护，并且可以减少类型错误的发生。在编写大型项目时，建议使用类型注解和函数注解来规范代码风格。