python joblib库调用
时间: 2023-12-15 17:03:43 浏览: 85
Python-joblib使用Python方便的进行并行计算
以下是使用joblib库的一个简单例子:
```python
from joblib import Parallel, delayed
def square(x):
return x**2
if __name__ == '__main__':
inputs = [1, 2, 3, 4, 5]
results = Parallel(n_jobs=2)(delayed(square)(i) for i in inputs)
print(results)
```
这个例子中,我们定义了一个函数`square`,用于计算输入参数的平方。然后,我们使用joblib库的`Parallel`和`delayed`函数来并行计算输入列表中每个元素的平方。`n_jobs`参数指定了使用的CPU核心数量。最后,我们打印出计算结果。
阅读全文