在Python中,可以使用类型提示来指定函数参数和返回值的类型。对于泛型数值类型的输入提示,可以使用TypeVar来定义一个泛型类型变量,然后在函数参数或返回值中使用该泛型类型变量。
下面是一个示例代码:
from typing import TypeVar
T = TypeVar('T', int, float)
def square(num: T) -> T:
return num * num
result = square(5) # 输入整数
print(result) # 输出:25
result = square(3.14) # 输入浮点数
print(result) # 输出:9.8596
在上面的代码中,我们使用TypeVar定义了一个泛型类型变量T,它可以是int或float类型。然后在函数square的参数和返回值中使用了这个泛型类型变量T。这样,在调用square函数时,可以传入整数或浮点数,并且函数返回的类型也会根据输入的类型进行推断。
这种方式可以提供输入提示,帮助开发者在编写代码时更加准确地指定参数和返回值的类型,从而提高代码的可读性和可维护性。
腾讯云相关产品和产品介绍链接地址:
领取专属 10元无门槛券
手把手带您无忧上云