Python中实现多线程 Threading 和多进程 Multiprocessing

开发 后端
最近正在和 ray 以及 spark 打交道,索性讲一下并发和并行。反正大家都是管理学院的,平时很少接触这种,因此这个选题不大可能因为内容基础而贻笑大方。

[[405163]]

本文转载自微信公众号「Piper蛋窝」,作者Piper蛋 。转载本文请联系Piper蛋窝公众号。

昨天晚上组会轮到我汇报技术内容,最近正在和 ray 以及 spark 打交道,索性讲一下并发和并行。反正大家都是管理学院的,平时很少接触这种,因此这个选题不大可能因为内容基础而贻笑大方。

本文摆一摆并发和并行。附上很简单的 Python 代码,涉及到自带库threading[1] 和 multiprocessing[2] 的使用。

并发和并行

咱们简单用多线程对应并发,多进程对应并行。多线程并发更强调充分利用性能;多进程并行更强调提升性能上限。

我用非常简单且不那么严谨的比喻来说明。

多线程

一个 CPU 相当于一个学生。

一个学生一周开一次组会,换句话说一周给老师汇报一次工作。

老师一般会给学生同时布置几个任务,比如做比赛、做项目、读论文,学生可能周一做做比赛、周二读读论文、周三做做项目... 到了组会,他就把三件事都拿出来汇报,老师很欣慰,因为在老师的视角里:学生这三件事是同时在做的。

多线程也是同一个道理,假设你的手机只有一块单核 CPU 。你的 CPU 这 0.01 秒用来播放音乐,下 0.01 秒用来解析网页... 在你的视角里:播放音乐和解析网页是同时进行的。你大可以畅快地边听音乐边网上冲浪

何谓充分利用性能? 如果这学生只有一项工作,那他这一周可能只需要花费两天来做任务,剩下时间摸鱼(针不搓,三点钟饮茶先!)。因此,我们用「多线程」来让学生实现『并发』,充分利用学生能力。

 

在实际情况中,多线程、高并发这些词语更多地出现在服务端程序里。比如一个网络连接由一个线程负责,一块 CPU 可以负责处理多个异步的请求,大大提升了 CPU 利用率。

多进程

多个 CPU ( CPU 的多核)相当于多个学生。

一个任务可以拆成几个任务相互协作、同时进行,则是多进程。

比如研究生课程,老师非得留个论文作业,都研究生了我去,留啥大作业。

那咱就多线程并行搞呗。确定了大概思路,剩下的一股脑写就行。咱队伍里一共甲乙丙丁四名同学,那就:

  • 甲同学负责 Introduction
  • 乙同学负责 Background
  • 丙同学负责 Related Works
  • 丁同学负责 Methodology

这是乙同学提出异议:不应该是先完成 Introduction 再写 Background ,一个个来嘛?

大哥,都研究生了嗷,作业糊弄糊弄差不多得了啊。让你写你就写。

可以预知,上述四部分同时进行,怎么也比一个人写四块要快。

 

所以说 多进程并行提升性能上限 。

在实际情况中,多进程更多地与高性能计算、分布式计算联系在一起。

Python 实现

首先声明咱的实验环境。

  1. > python --version 
  2. Python 3.8.5 

咱们设置个任务:求数的欧拉函数值。

  1. def euler_func(n: int) -> int
  2.     res = n 
  3.     i = 2 
  4.     while i <= n // i: 
  5.         if n % i == 0: 
  6.             res = res // i * (i - 1) 
  7.             while (n % i == 0): n = n // i 
  8.         i += 1 
  9.     if n > 1: 
  10.         res = res // n * (n - 1) 
  11.     return res 

求一个数的欧拉函数值可能很快,但是一堆数呢?

所以咱想着用并行完成这个任务。

咱们把任务分成三份。

  1. task1 = list(range(2, 50000, 3))  # 2, 5, ... 
  2. task2 = list(range(3, 50000, 3))  # 3, 6, ... 
  3. task3 = list(range(4, 50000, 3))  # 4, 7, ... 
  4.  
  5. def job(task: List): 
  6.     for t in task: 
  7.         euler_func(t) 

来看看平平无奇的正常串行。

  1. @timer 
  2. def normal(): 
  3.     job(task1) 
  4.     job(task2) 
  5.     job(task3) 

完成了 task1 再完成 task2 ... 行,没毛病。

看看多线程?

  1. import threading as th 
  2.  
  3. @timer 
  4. def mutlthread(): 
  5.     th1 = th.Thread(target=job, args=(task1, )) 
  6.     th2 = th.Thread(target=job, args=(task2, )) 
  7.     th3 = th.Thread(target=job, args=(task3, )) 
  8.  
  9.     th1.start() 
  10.     th2.start() 
  11.     th3.start() 
  12.  
  13.     th1.join() 
  14.     th2.join() 
  15.     th3.join() 

再看看多进程?

  1. import multiprocessing as mp 
  2.  
  3. @timer 
  4. def multcore(): 
  5.     p1 = mp.Process(target=job, args=(task1, )) 
  6.     p2 = mp.Process(target=job, args=(task2, )) 
  7.     p3 = mp.Process(target=job, args=(task3, )) 
  8.  
  9.     p1.start() 
  10.     p2.start() 
  11.     p3.start() 
  12.  
  13.     p1.join() 
  14.     p2.join() 
  15.     p3.join() 

上述代码的逻辑是这样的:

  • 我创建线程/进程,其生来的目的就是完成任务job(task1)或job(task2)、job(task3),注意这里函数名和参数被分开了target=job, args=(task1, )
  • 然后 start() ,告诉线程/进程:你可以开始干活了
  • 他们自己干自己的,咱们程序主逻辑还得继续往下运行
  • 到 join() 这里,咱们是指让线程/进程阻塞住咱的主逻辑,比如p1.join()是指:p1不干完活,我主逻辑不往下进行(属于是「阻塞」)
  • 这样,我们的函数multcore结束后,一定其中的线程/进程任务都完成了

咱看看结果:

  1. if __name__ == '__main__'
  2.  
  3.     print("同步串行:"
  4.     normal() 
  5.  
  6.     print("多线程并发:"
  7.     mutlthread() 
  8.  
  9.     print("多进程并行:"
  10.     multcore() 
  11.  
  12. # 下面是结果 
  13. 同步串行: 
  14. timer: using 0.24116 s 
  15. 多线程并发: 
  16. timer: using 0.24688 s 
  17. 多进程并行: 
  18. timer: using 0.13791 s 

结果不太对,按理说,多进程并行的耗时应该是同步串行的三分之一,毕竟三个同等体量的任务在同时进行。

多线程并发比同步串行慢是应该的,因为多线程并发和同步串行的算力是一样的,但是多线程并发得在各个任务间来回切换,导致更慢。

你问 @timer 是什么意思?哦,这个是我写的修饰器,如下。

  1. def timer(func): 
  2.     @wraps(func) 
  3.     def inner_func(): 
  4.         t = time.time() 
  5.         rts = func() 
  6.         print(f"timer: using {time.time() - t :.5f} s"
  7.         return rts 
  8.     return inner_func 

 

不太明白『Python修饰器』的老铁,不如给我点个「在看」,再关注下我,咱们以后详细道来。

 

责任编辑:武晓燕 来源: Piper蛋窝
相关推荐

2021-04-20 12:39:52

Node.js多线程多进程

2022-03-09 17:01:32

Python多线程多进程

2021-08-04 23:30:28

Node.js开发线程

2023-12-11 18:18:24

Python编程线程

2023-11-28 13:52:00

Python多进程多线程

2020-04-07 11:10:30

Python数据线程

2019-02-26 11:15:25

进程多线程多进程

2024-01-15 10:55:40

Python多线程开发

2017-01-10 13:39:57

Python线程池进程池

2010-07-26 09:45:09

Perl多进程

2023-12-13 09:56:13

​多进程多线程协程

2023-05-10 07:47:08

Python并发编程

2023-11-01 11:20:57

2016-10-09 20:15:30

多线程多进程

2022-01-04 09:53:37

Python多线程多进程

2021-09-10 21:25:43

Redis分布式

2013-12-02 17:33:20

Linux进程多线程

2019-08-12 16:22:07

Python线程场景

2017-06-30 10:12:46

Python多进程

2021-06-11 11:28:22

多线程fork单线程
点赞
收藏

51CTO技术栈公众号