python中的泛型使用TypeVar
1.引入为什么需要TypeVar
PEP484的作者希望借助typing模块引入类型提示,不改动语言的其它部分。通过精巧的元编程技术,让类支持[]运算不成问题。但是方括号内的T变量必须在某处定义,否则要大范围改动python解释器才能让泛型支持特殊的[]表示法。
鉴于此,我们增加了typing.TypeVar构造函数,把变量名称引入当前命名空间。
由于java,c#和TypingScript等语言不要求实现声明类型变量的名称,因此没有与python的TypeVar类对应的结构。
2.示例
from collections.abc import Iterable
from typing import TypeVar
T = TypeVar('T')
def mode(data: Iterable[T]) -> T:
return data[1:]
if __name__ == '__main__':
list1 = [1.1,1.2,1.3,1.4,1.5]
list2 = [1,2,3,4,5]
data1 = mode(list1)
data2 = mode(list2)
print(data1)
print(data2)
3.受限的TypeVar
TypeVar还接受一些位置参数,以对类型参数施加限制。
from collections.abc import Iterable
from decimal import Decimal
from fractions import Fraction
from typing import TypeVar
NumberT = TypeVar('NumberT', float, Decimal, Fraction)
def mode(data: Iterable[NumberT]) -> NumberT:
...
4.有界的TypeVar
from collections.abc import Iterable, Hashable
def mode(data: Iterable[Hashable]) -> Hashable:
现在的问题是,返回的项是Hashable类型。Hashable是一个抽象基类。只实现了__hash__方法。因此,除了调用hash(),类型检查工具不会允许对返回值做其它任何操作。
所以,这么做没有任何意义。解决方法是使用TypeVar的另一个可选参数,即关键字参数bound。这个参数会为可接受的类型设定一个上边界。
下面的实例使用bound=Hashable指明,类型参数可以是Hashable或它的任何子类型。
from collections import Counter
from collections.abc import Iterable, Hashable
from typing import TypeVar
#Python学习交流群:711312441
HashableT = TypeVar('HashableT', bound=Hashable)
def mode(data: Iterable[HashableT]) -> HashableT:
pairs = Counter(data).most_common(1)
if len(pairs) == 0:
raise ValueError('no mode for empty data')
return pairs[0][0]