![53ba23a639bdae96e7fa05bc9fc46fa3.png](https://i-blog.csdnimg.cn/blog_migrate/5e9ac8feff06845db7996951aa5006b1.jpeg)
学过python的同学都知道,python中没有类型声明的语句,所以经常搞不清要给函数传什么类型的参数,比如下面经典的歧义函数:
def add(x, y):
return x+y
在没有任何说明的情况下,很难搞清楚x和y分别表示什么,可以是两个数值或字符串:
print(add(2, 4))
print(add("w", "t"))
没有具体的参数类型这种写法能起到类似多态的作用,使得函数不必指定具体的对象类型,增加了灵活性。但是,如果希望自己写的函数参数类型明确,应该怎么做呢?根据相关资料整理得到下面的几种策略。
策略1:使用__annotations__注解
在函数形参的名字后面加上“:注解”,可以实现函数注解【1】功能:
def add(x:int, y:int):
return x+y
对于较新的python版本可以直接使用函数的__annotations__属性查看注解内容:
print(add.__annotations__)
将会得到一个包含add的参数名称和对应类型说明的字典:
{'x': <class 'int'>, 'y': <class 'int'>}
策略2:使用isinstance进行强制类型检查
python提供的isinstance(obj, class)函数可以检查obj是不是class的对象,比如强制检查add函数中x是不是int类型:
def add(x, y):
# 强制类型检查
if not isinstance(x, int):
raise ValueError("x is not int")
if not isinstance(y, int):
raise ValueError("y is not int")
return x+y
现在,add函数传入的参数x和y就必须是int类型,否则就会报错。
参考:
【1】https://www.python.org/dev/peps/pep-3107/
【2】https://www.jianshu.com/p/7a2c9133a002