Python如何定义子函数的参数和返回值
文章目录
在Python中,可以使用类型注解来约束输入和输出变量的类型。类型注解是一种标记代码,用于指定变量的类型。
在函数定义中,可以使用->
符号来指定函数返回值的类型,例如:
def add(a: int, b: int) -> int:
return a + b
在上面的例子中,add
函数接受两个整数类型的参数a
和b
,并返回一个整数类型的值。如果函数的返回值不是整数类型,则会引发TypeError异常。
对于输入参数,可以使用类型注解来指定其类型。例如,如果要求输入参数必须是整数类型,可以使用以下代码:
def add(a: int, b: int) -> int:
if not isinstance(a, int) or not isinstance(b, int):
raise TypeError("a and b must be integers")
return a + b
在上面的例子中,isinstance
函数用于检查输入参数a
和b
是否为整数类型。如果不是整数类型,则引发TypeError异常。
需要注意的是,类型注解并不会自动执行类型检查。它们只是提供给程序员的一种标记,以便在代码中更清楚地说明变量和函数的类型要求。因此,程序员仍然需要自己编写代码来执行类型检查。