python - 我可以将 map/imap/imap_unordered 与不带参数的函数一起使用吗？

Question

有时我需要对不带参数的函数使用多处理。我希望我能做类似的事情:

from multiprocessing import Pool

def f():  # no argument
    return 1

# TypeError: f() takes no arguments (1 given)
print Pool(2).map(f, range(10))

我可以做 Process(target=f, args=())，但我更喜欢 map/imap/imap_unordered。有办法吗？

Answer 1

您可以像这样使用 pool.starmap() 而不是 .map():

from multiprocessing import Pool

def f():  # no argument
    return 1

print Pool(2).starmap(f, [() for _ in range(10)])

starmap 会将给定可迭代对象的所有元素作为参数传递给 f。在你的情况下，迭代应该是空的。