python - 如何使用带条件的地图功能？

Question

我有以下循环：

for i in data.id.unique():
    if i in group.groups.keys():
      main_func(i)

如何在 python 中使用 map 函数进行多处理，我想在 Pool 函数中应用上面的代码

pool = Pool(processes=5)
pool.map(main_func, group.groups.keys())

Answer 1

我认为你可以使用这个：

pool = Pool(processes=5)
pool.map(main_func, [i for i in data.id.unique() if i in group.groups.keys()])

Answer 2

如何使用 lambda：

pool.map(lambda x: main_func(x) if x in groups.groups.keys() else None, data.id.unique())