mypy 类型和可选过滤

问题描述

我是 mypy 的新手,我在从集合中过滤选项时找不到正确的方法来使类型正常工作。

from typing import Optional,List

list1: List[Optional[int]] = [1,None,3]
# This type is an error,filter can't even handle a list with optionals
list2: List[int] = list(filter(lambda n: n is not None,list1))

我应该投这个吗? mypy 应该能够推断出 None 现在被过滤掉了吗?

解决方法

使用 typing.cast 应该可行,但我认为这里最好的解决方案是使用列表理解。

以下代码为我传递了 mypy:

from typing import Optional,List

list1: List[Optional[int]] = [1,None,3]
list2: List[int] = [x for x in list1 if x is not None]

一般来说,列表推导式是表达映射/过滤操作的更惯用的方式。 (Guido 实际上想在 Python 3 中删除 map/filter!)