python多进程

1、前言

  python中使用多进程，可以加快代码的运行速度，更高效地进行相关工作。

2、示例

  使用蒙特卡洛方法计算$\pi$来进行使用多进程前后代码运行速率的对比；

import random
import multiprocessing as mp
from tqdm import trange
from joblib import Parallel, delayed

def cal_pi(n):
    t = 0
    for _ in range(n):
        x = random.random()
        y = random.random()
        if x**2 + y**2 <= 1:
            t += 1
    return t / n * 4

• 不使用多进程

%%time
pi = 0
for i in range(10):
    pi += cal_pi(100_0000)

print(pi/10)

• 使用多进程

%%time
# ret = mp.Pool(4).map(cal_pi, [100_000]*10)
ret = Parallel(n_jobs=4)(delayed(cal_pi)(100_0000) for i in range(10))

3、参考

用一行代码实现python多进程加速