使用CPU多核进行并行运算

import multiprocessing

def function(x):
    return(x * x)
    
arg_list = [1, 3, 5, 7]

cores = multiprocessing.cpu_count()
pool = multiprocessing.Pool(processes=cores)
for result in pool.imap(function, arg_list):
    print(result)

这样定义好了函数和参数列表，就可以轻松进行CPU多核并行运算。