使用 Dask 在 Python 中进行并行计算 | Linux 中国
发布日期:2021-05-06 14:57:08 浏览次数:40 分类:原创文章

本文共 1197 字,大约阅读时间需要 3 分钟。

640?wx_fmt=jpegDask 库可以将 Python 计算扩展到多个核心甚至是多台机器。-- Moshe Zadka

关于 Python 性能的一个常见抱怨是全局解释器锁[1](GIL)。由于 GIL,同一时刻只能有一个线程执行 Python 字节码。因此,即使在现代的多核机器上,使用线程也不会加速计算。

但当你需要并行化到多核时,你不需要放弃使用 Python:Dask[2] 库可以将计算扩展到多个内核甚至多个机器。某些设置可以在数千台机器上配置 Dask,每台机器都有多个内核。虽然存在扩展规模的限制,但一般达不到。

虽然 Dask 有许多内置的数组操作,但举一个非内置的例子,我们可以计算偏度[3]

  
  1. import numpy
  2. import dask
  3. from dask import array as darray
  4. arr = dask.from_array(numpy.array(my_data), chunks=(1000,))
  5. mean = darray.mean()
  6. stddev = darray.std(arr)
  7. unnormalized_moment = darry.mean(arr * arr * arr)
  8. ## See formula in wikipedia:
  9. skewness = ((unnormalized_moment - (3 * mean * stddev ** 2) - mean ** 3) /
  10. stddev ** 3)

请注意,每个操作将根据需要使用尽可能多的内核。这将在所有核心上并行化执行,即使在计算数十亿个元素时也是如此。

当然,并不是我们所有的操作都可由这个库并行化,有时我们需要自己实现并行性。

为此,Dask 有一个“延迟”功能:

  
  1. import dask
  2. def is_palindrome(s):
  3. return s == s[::-1]
  4. palindromes = [dask.delayed(is_palindrome)(s) for s in string_list]
  5. total = dask.delayed(sum)(palindromes)
  6. result = total.compute()

这将计算字符串是否是回文并返回回文的数量。

虽然 Dask 是为数据科学家创建的,但它绝不仅限于数据科学。每当我们需要在 Python 中并行化任务时,我们可以使用 Dask —— 无论有没有 GIL。


via: https://opensource.com/article/19/4/parallel-computation-python-dask

作者:Moshe Zadka (Community Moderator)[5] 选题:lujun9972 译者:geekpi 校对:wxy

本文由 LCTT 原创编译,Linux中国 荣誉推出

640?wx_fmt=png
上一篇:使用 RadioDroid 流传输网络广播 | Linux 中国
下一篇:破除对 AWS Fargate 的幻觉 | Linux 中国

发表评论

最新留言

路过,博主的博客真漂亮。。
[***.116.15.85]2025年04月11日 15时34分49秒