我正在使用 map 处理 Python3.6 中的列表:
def calc(num):
if num > 5:
return None
return num * 2
r = map(lambda num: clac(num), range(1, 10))
print(list(r))
# => [2, 4, 6, 8, 10, None, None, None, None]
我期望的结果是:[2, 4, 6, 8, 10]。
当然,我可以使用filter 来处理map 结果。但是有没有办法让map直接返回我想要的结果呢?
请您参考如下方法:
map 不能直接过滤掉items。它为每个输入项目输出一个项目。您可以使用列表理解从结果中过滤掉 None。
r = [x for x in map(calc, range(1,10)) if x is not None]
(这只对范围内的每个数字调用一次 calc。)
旁白:无需编写 lambda num: calc(num)。如果您想要一个返回 calc 结果的函数,只需使用 calc 本身。
