import multiprocessing
from multiprocessing import Pool# 导入 Pool 允许你创建一个进程池
# 进程池是一组工作进程,它们可以并行地执行多个任务# multiprocessing.cpu_count(): 返回当前机器上的CPU核心数
sum_cpu = multiprocessing.cpu_count()use_cpu = max(1, int(multiprocessing.cpu_count() * 0.5))
# 计划使用的CPU核心数# 定义要并行执行的函数
def square(number):return number * number# 要处理的数字列表
numbers = [1, 2, 3, 4, 5]# 创建进程池,并使用imap方法并行执行square函数
if __name__ == '__main__':with Pool(processes=4) as pool: # 假设我们使用4个进程squared_numbers = pool.imap(square, numbers) # 列表numbers中的每一个元素,作为并行执行函数的参数输入# 打印每个数字的平方for squared_number in squared_numbers:print(squared_number)