在Python3的map函数中排除空值

Question

我正在使用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直接返回到我想要的结果？

Answer 1

使用map本身时不需要，但您可以将map（）调用更改为：

r = [calc(num) for num in range(1, 10) if calc(num) is not None]
print(r)  # no need to wrap in list() anymore

才能得到你想要的结果。

Answer 2

map不能直接过滤掉项目。它为每项输入输出一项。您可以使用列表信息从结果中过滤掉No。

r = [x for x in map(calc, range(1,10)) if x is not None]

（这只对范围内的每个数字调用calc一次。）

旁白：没有必要写lambda num： calc（num）。如果您想要一个返回calc结果的函数，只需使用calc本身。