将python的多处理模块与素数函数相结合

Question

我正在使用 python 多处理/池模块来比较检查数字是否为素数的函数的程序运行时间。我可以让主要功能自行工作，但似乎无法将两者合并，以便它们一起工作。池函数如下：

import math import time
import multiprocessing from multiprocessing import Pool

def pool_process(f,data,pool_size):

    tp1 = time.time()
    pool = Pool(processes = pool_size)
    result = pool.map(f, data)
    pool.close()
    pool.join()
    #enter code here
    print("Results", result)
    print("Overall Time:", (time.time()-tp1))

我用来检查一个数是否为素数的函数如下：

def check_prime(num):

    t1 = time.time()
    res = False

    if num > 0:

        for i in range(2,num):
            if (num % i) == 0:
                print(num,"is not a prime number")
                print(i,"times",num//i,"is",num)
                print("Time:", (time.time()-t1))
                break
        else:
            print(num,"is a prime number")
            print("Time:", time.time()-t1) 
            res = True

    return res

然后我调用 pool_process 函数

data = 15488801
pool_process(check_prime, data, 1)

我收到以下错误：

'int' object is not iterable

Answer 1

第三个代码sn-p中的data需要是一个列表。

变量data 被输入pool_process 作为第二个参数。

data = [15488801]
pool_process(check_prime, data, 1)

在那里它被提供给Pool.map作为迭代的参数。

def pool_process(f,data,pool_size):
    # ...
    result = pool.map(f, data)

在我将该整数放入列表（并更正import 语句）后，代码在我的机器上运行良好。