python - 递归函数值生成器的多重处理

Question

我有一个简单的递归函数及其值生成器:

def rec(n):
    if n == 1:
        return 1
    else:
        return n + rec(n-1)

lis=[]    
for i in range(1,1000):
    lis.append(rec(i))

输出:

[1, 3, 6, 10, 15, 21, 28, 36,...]

在这样的构造中可以使用多处理吗？

Answer 1

您在这里描述的是干净 map :将一个可迭代的值映射到一个值列表，使得原始列表中的 x_i 映射到 f(x_i) 在结果中。

我们可以将它与池一起使用:

from multiprocessing import Pool

def rec(n):
    if n == 1:
        return 1
    else:
        return n + rec(n-1)

pool = Pool(4)

lis = pool.map(rec, range(1,1000))

这里4是处理请求的进程数。通常这是核心数(或核心数的倍数)。

但请注意，在您的情况下，您可以将 rec 函数替换为:

# fast implementation
def rec(n):
    return n*(n+1)//2