欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

python实现多进程代码示例

程序员文章站 2022-04-28 17:46:22
想要充分利用多核cpu资源,python中大部分情况下都需要使用多进程,python中提供了multiprocessing这个包实现多进程。multiprocessing支...

想要充分利用多核cpu资源,python中大部分情况下都需要使用多进程,python中提供了multiprocessing这个包实现多进程。multiprocessing支持子进程、进程间的同步与通信,提供了process、queue、pipe、lock等组件。

开辟子进程

multiprocessing中提供了process类来生成进程实例

process([group [, target [, name [, args [, kwargs]]]]])

  • group分组,实际上不使用
  • target表示调用对象,你可以传入方法的名字
  • args表示给调用对象以元组的形式提供参数,比如target是函数a,他有两个参数m,n,那么该参数为args=(m, n)即可
  • kwargs表示调用对象的字典
  • name是别名,相当于给这个进程取一个名字

先来个小例子:

# -*- coding:utf-8 -*-
from multiprocessing import process, pool
import os
import time


def run_proc(wtime):
  n = 0
  while n < 3:
    print "subprocess %s run," % os.getpid(), "{0}".format(time.ctime())  #获取当前进程号和正在运行是的时间
    time.sleep(wtime)  #等待(休眠)
    n += 1

if __name__ == "__main__":
  p = process(target=run_proc, args=(2,)) #申请子进程
  p.start()   #运行进程
  print "parent process run. subprocess is ", p.pid
  print "parent process end,{0}".format(time.ctime())

运行结果:

parent process run. subprocess is 30196
parent process end,mon mar 27 11:20:21 2017
subprocess 30196 run, mon mar 27 11:20:21 2017
subprocess 30196 run, mon mar 27 11:20:23 2017
subprocess 30196 run, mon mar 27 11:20:25 2017

根据运行结果可知,父进程运行结束后子进程仍然还在运行,这可能造成僵尸( zombie)进程。

通常情况下,当子进程终结时,它会通知父进程,清空自己所占据的内存,并在内核里留下自己的退出信息。父进程在得知子进程终结时,会从内核中取出子进程的退出信息。但是,如果父进程早于子进程终结,这可能造成子进程的退出信息滞留在内核中,子进程成为僵尸(zombie)进程。当大量僵尸进程积累时,内存空间会被挤占。

有什么办法可以避免僵尸进程呢?

这里介绍进程的一个属性 deamon,当其值为true时,其父进程结束,该进程也直接终止运行(即使还没运行完)。
所以给上面的程序加上p.deamon = true,看看效果。

# -*- coding:utf-8 -*-
from multiprocessing import process, pool
import os
import time


def run_proc(wtime):
  n = 0
  while n < 3:
    print "subprocess %s run," % os.getpid(), "{0}".format(time.ctime())
    time.sleep(wtime)
    n += 1

if __name__ == "__main__":
  p = process(target=run_proc, args=(2,))
  p.daemon = true  #加入daemon
  p.start()
  print "parent process run. subprocess is ", p.pid
  print "parent process end,{0}".format(time.ctime())

执行结果:

parent process run. subprocess is 31856
parent process end,mon mar 27 11:40:10 2017

这是问题又来了,子进程并没有执行完,这不是所期望的结果。有没办法将子进程执行完后才让父进程结束呢?

这里引入p.join()方法,它使子进程执行结束后,父进程才执行之后的代码

# -*- coding:utf-8 -*-
from multiprocessing import process, pool
import os
import time


def run_proc(wtime):
  n = 0
  while n < 3:
    print "subprocess %s run," % os.getpid(), "{0}".format(time.ctime())
    time.sleep(wtime)
    n += 1

if __name__ == "__main__":
  p = process(target=run_proc, args=(2,))
  p.daemon = true
  p.start()
  p.join()  #加入join方法
  print "parent process run. subprocess is ", p.pid
  print "parent process end,{0}".format(time.ctime())

执行结果:

subprocess 32076 run, mon mar 27 11:46:07 2017
subprocess 32076 run, mon mar 27 11:46:09 2017
subprocess 32076 run, mon mar 27 11:46:11 2017
parent process run. subprocess is 32076
parent process end,mon mar 27 11:46:13 2017

这样所有的进程就能顺利的执行了。

将进程定义成类

通过继承process类,来自定义进程类,实现run方法。实例p通过调用p.start()时自动调用run方法。

如下:

# -*- coding:utf-8 -*-
from multiprocessing import process, pool
import os
import time


class myprocess(process):

  def __init__(self, wtime):
    process.__init__(self)
    self.wtime = wtime

  def run(self):
    n = 0
    while n < 3:
      print "subprocess %s run," % os.getpid(), "{0}".format(time.ctime())
      time.sleep(self.wtime)
      n += 1


if __name__ == "__main__":
  p = myprocess(2)
  p.daemon = true
  p.start()  #自动调用run方法
  p.join()
  print "parent process run. subprocess is ", p.pid
  print "parent process end,{0}".format(time.ctime())

执行结果和上一个例子相同。

创建多个进程

很多时候系统都需要创建多个进程以提高cpu的利用率,当数量较少时,可以手动生成一个个process实例。当进程数量很多时,或许可以利用循环,但是这需要程序员手动管理系统中并发进程的数量,有时会很麻烦。这时进程池pool就可以发挥其功效了。可以通过传递参数限制并发进程的数量,默认值为cpu的核数。

直接上例子:

# -*- coding:utf-8 -*-
from multiprocessing import process,pool
import os,time

def run_proc(name):    ##定义一个函数用于进程调用
  for i in range(5):  
    time.sleep(0.2)  #休眠0.2秒
    print 'run child process %s (%s)' % (name, os.getpid())
#执行一次该函数共需1秒的时间

if __name__ =='__main__': #执行主进程
  print 'run the main process (%s).' % (os.getpid())
  mainstart = time.time() #记录主进程开始的时间
  p = pool(8)      #开辟进程池
  for i in range(16):                 #开辟14个进程
    p.apply_async(run_proc,args=('process'+str(i),))#每个进程都调用run_proc函数,
                            #args表示给该函数传递的参数。

  print 'waiting for all subprocesses done ...'
  p.close() #关闭进程池
  p.join() #等待开辟的所有进程执行完后,主进程才继续往下执行
  print 'all subprocesses done'
  mainend = time.time() #记录主进程结束时间
  print 'all process ran %0.2f seconds.' % (mainend-mainstart) #主进程执行时间

执行结果:

开头部分

run the main process (30920).
waiting for all subprocesses done …
run child process process0 (32396)
run child process process3 (25392)
run child process process1 (28732)
run child process process2 (32436)

末尾部分:

run child process process15 (25880)
all subprocesses done
all process last 2.49 seconds.

相关说明:

这里进程池对并发进程的限制数量为8个,而程序运行时会产生16个进程,进程池将自动管理系统内进程的并发数量,其余进程将会在队列中等待。限制并发数量是因为,系统中并发的进程不是越多越好,并发进程太多,可能使cpu大部分的时间用于进程调度,而不是执行有效的计算。

采用多进程并发技术时,就单个处理机而言,其对进程的执行是串行的。但具体某个时刻哪个进程获得cpu资源而执行是不可预知的(如执行结果的开头部分,各进程的执行顺序不定),这就体现了进程的异步性。

如果单个程序执行14次run_proc函数,那么它会需要至少16秒,通过进程的并发,这里只需要2.49秒,可见并发的优势。

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。