我正在使用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
直接返回到我想要的结果呢?
发布于 2018-07-05 19:27:07
map
不能直接过滤掉项目。它为每一项输入输出一项。您可以使用列表合并从结果中过滤出None
。
r = [x for x in map(calc, range(1,10)) if x is not None]
(这仅对范围内的每个号码调用一次calc
。)
旁白:不需要编写lambda num: calc(num)
。如果您想要一个返回calc
结果的函数,只需使用calc
本身。
发布于 2018-07-05 19:06:10
使用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
得到你想要的结果。
https://stackoverflow.com/questions/51189698
复制相似问题