python-进程池判断指定范围的素数个数

# 菜鸟出没
# 非战斗人员请撤离
# 时间：2022/3/29 20:20
import multiprocessing
import math


def is_prime(m):
    """
    判断一个数m是不是素数
    :param m: 正整数
    """
    #        请在此处添加代码       #
    # *************begin************#
    if m <= 1:
        return False
    for j in range(2, int(math.sqrt(m)) + 1):
        if m % j == 0:
            return False
    return True

    # **************end*************#


def main(n):
    """
    判断0~n之间素数的个数
    :param m: 正整数
    """
    #        请在此处添加代码       #
    # *************begin************#
    pools = multiprocessing.Pool(5)
    print(sum(pools.map(is_prime, range(n))))
    # **************end*************#


if __name__ == '__main__':
    n = int(input())
    main(n)