我在一个列表中有两个类实例:
from typing import List, Union
class A:
my_type: str = 'A'
class B:
my_type: str = 'B'
my_list: List[Union[A, B]] = [A(),A(),B(),B()]
As: List[A] = [a for a in my_list if a.my_type == 'A']
def function_that_gets_only_array_of_As(arr: List[A]):
print(arr)
function_that_gets_only_array_of_As(As) # this yields a type hint error
我将如何暗示 As 是列表[A] 类型?
可以使用类型断言/强制转换来重写推断的类型:
from typing import cast
As = cast(List[A], [...])
基本上由你来确保你的类型是你声明的。