多处理列表理解/过滤器

Question

我正在尝试对列表理解进行多处理：

l = [i for i in range(10) if fun(i)]

如果我保留列表中的所有值，我可以简单地调用pool.map ，但此循环可能不会保留所有值。

我怎样才能 go 实现这个？

Answer 1

使用 itertools.compress 过滤 pool.map 的结果。

这是使用过滤器的串行实现：

ages = [5, 12, 17, 18, 24, 32]

def fun(x):
  if x < 18:
    return False
  else:
    return True

adults = list(filter(fun, ages))

adults
  
Out[29]: [18, 24, 32]

并行实现使用 map 并压缩：

from itertools import compress
import multiprocessing
with multiprocessing.Pool(processes=2) as pool:
    adult_ages = list(compress(ages, pool.map(fun, ages)))
adult_ages

Out[27]: [18, 24, 32]