joblib怎么使用进程池?
时间: 2024-02-17 16:35:26 浏览: 138
Python多进程并发(multiprocessing)用法实例详解
5星 · 资源好评率100%
使用 `joblib` 库实现进程池非常简单,只需要在调用 `Parallel` 函数时指定 `n_jobs` 参数即可。例如,下面的代码将使用 4 个进程执行 `my_function` 函数。
```python
from joblib import Parallel, delayed
def my_function(x):
# 一些长时间运行的任务
return x * x
results = Parallel(n_jobs=4)(delayed(my_function)(i) for i in range(10))
print(results)
```
在这个例子中,`n_jobs` 参数设置为 4,这意味着 `my_function` 函数将在 4 个不同的进程中并行运行。`delayed` 函数用于将函数调用包装成可被并行化的对象。在这个例子中,我们使用了一个简单的函数 `my_function`,该函数接受一个参数并返回该参数的平方。
`Parallel` 函数的返回值是一个列表,其中包含了每个函数调用的结果。在这个例子中,我们打印了该列表,它应该包含了 0 到 9 的平方值。
阅读全文