python多进程

当有多个非相关任务需要处理时,并行能大大提高处理速度。这里简要介绍python的multiprocessing模块。

各种方法是否是阻塞的可参考 python语法——使用Pool实现多进程并行

map() 会使进程阻塞,即通过 map() 开启的多进程都结束之后,这个函数才会有返回结果,否则主进程会一直等待,不会往下进行 。
map_async() 为非阻塞,即通过 map_async() 开启多进程之后,立刻会返回结果,主进程会继续往下执行。

注意:如果后面调用了 join() 函数,则不管之前用的是 map 还是 map_async,主进程都会等待,直到进程池中所有进程执行完毕,才会继续往下执行。


简单多进程编写

当我们任务数量确定而且比较少的时候,可以手动为每个任务指定一个进程来运行。

import multiprocessing as mp

def f(a):
    print(a)

if __name__ == '__main__':
    # 这里有三个任务,手动指定3个进程
    p1 = mp.Process(target=f, args=(1,))
    p2 = mp.Process(target=f, args=(2,))
    p3 = mp.Process(target=f, args=(3,))
    p1.start()
    p2.start()
    p3.start()
    p1.join()
    p2.join()
    p3.join()

# 输出
1
2
3

使用进程池来处理多任务

当我们任务比较多而且不确定数量(又或者想使得代码更简洁)的时候可以使用进程池Pool来编写多进程。

import multiprocessing as mp

def f(a):
    return a

if __name__ == '__main__':
    pool = mp.Pool()
    res = pool.map(f, (1, 2, 3, 4))
    print(res)
# 输出
[1, 2, 3, 4]

可以看到,进程池不仅可以方便的处理多进程,还将各个进程的的处理结果储存了在一个列表中。Pool默认使用计算机所有cpu核来进行运算,也可使用Pool(process=4)来指定并行的进程数。
另一个可以储存结果的函数是apply_async(),但是这个函数只支持传入一个参数,也即运行一个任务,运行多个任务需要多次指定。此外,他返回的结果是一个类似生成器的东西,需要通过get函数取出来。

import multiprocessing as mp

def f(a):
    return a

if __name__ == '__main__':
    pool = mp.Pool()
    res = [pool.apply_async(f, (i,)) for i in range(4)]
    print([r.get() for r in res])

# 输出
[0, 1, 2, 3]

多参数函数的多进程

上面处理任务的函数都只有一个参数,在实际情况这种情况是很少的,一般我们的函数都需要传入多个参数。python2中未能支持传多个参数,python3.3后则有starmap来支持传入多参数。

import multiprocessing as mp

def f(a, b):
    return (a, b)

if __name__ == '__main__':
    pool = mp.Pool()
    res = pool.starmap(f, ((1, 2), ('a', 'b')))
    print(res)

# 输出
[(1, 2), ('a', 'b')]

另一个传多个参数的方法是通过python中的魔术方法*args:

import multiprocessing as mp

def f(*l):
    a, b = l[0]  # 为什么这里是l[0]而不是l?因为这里传入的l是一个嵌套元组((a,b),)。
    return (a, b)

if __name__ == '__main__':
    pool = mp.Pool()
    res = pool.map(f, ((1, 2), ('a', 'b')))
    print(res)

# 输出
[(1, 2), ('a', 'b')]

亦或使用**kwargs:

import multiprocessing as mp

def f(*l):
    d = l[0]    # 这里l为({'a': 1, 'b': 2},)
    return (d['a'], d['b'])

if __name__ == '__main__':
    pool = mp.Pool()
    res = pool.map(f, ({'a': 1, 'b': 2}, {'a': 3, 'b': 4}))
    print(res)

# 输出
[(1, 2), (3, 4)]

注:需要注意的是mapstarmap都是主进程阻塞的,无论是否调用join()函数。

共享内存和进程锁

一般情况下,各个进程中的数据变量是无法发生交流的,但我们可以通过使用Value数据存储在一个共享的内存表中。

import multiprocessing as mp
import time


def job(v, num):
    for _ in range(5):
        time.sleep(0.1)  # 暂停0.1秒,让输出效果更明显
        v.value += num  # v.value获取共享变量值
        print(v.value)


def multicore():
    v = mp.Value('i', 0)  # 定义共享变量
    p1 = mp.Process(target=job, args=(v, 1))
    p2 = mp.Process(target=job, args=(v, 3))  # 设定不同的number看如何抢夺内存
    p1.start()
    p2.start()
    p1.join()
    p2.join()


if __name__ == '__main__':
    multicore()

# 输出
1
5
9
13
17
4
8
12
16
20

在上面的代码中,我们定义了一个共享变量v,两个进程都可以对它进行操作。 在job()中我们想让v每隔0.1秒输出一次累加num的结果,但是在两个进程p1和p2 中设定了不同的累加值。很明显可以看到他们发生了冲突。

为了解决上述不同进程抢共享资源的问题,我们可以用加进程锁来解决。

import multiprocessing as mp


def job(v, num, l):
    l.acquire()  # 锁住
    for _ in range(5):
        time.sleep(0.1)
        v.value += num  # 获取共享内存
        print(v.value)
    l.release()  # 释放


def multicore():
    l = mp.Lock()  # 定义一个进程锁
    v = mp.Value('i', 0)  # 定义共享内存
    p1 = mp.Process(target=job, args=(v, 1, l))  # 需要将lock传入
    p2 = mp.Process(target=job, args=(v, 3, l))
    p1.start()
    p2.start()
    p1.join()
    p2.join()


if __name__ == '__main__':
    multicore()

# 输出
3
6
9
12
15
16
17
18
19
20

需要主义的是上面可能仍然会发生冲突——p1先执行还是p2先执行的问题。为了解决这个问题我们可以在start,join中决定他们的顺序。

import multiprocessing as mp

def job(v, num, l):
    l.acquire()  # 锁住
    for _ in range(5):
        time.sleep(0.1)
        v.value += num  # 获取共享内存
        print(v.value)
    l.release()  # 释放

def multicore():
    l = mp.Lock()  # 定义一个进程锁
    v = mp.Value('i', 0)  # 定义共享内存
    p1 = mp.Process(target=job, args=(v, 1, l))  # 需要将lock传入
    p2 = mp.Process(target=job, args=(v, 3, l))
    p1.start()
    p1.join()
    p2.start()
    p2.join()

if __name__ == '__main__':
    multicore()

# 输出
1
2
3
4
5
8
11
14
17
20

对于start和join方法:

  • start方法表示进程开始运行;
  • join表示主进程阻塞,等待子进程完成;

关于守护进程

1、守护进程会在主进程代码执行结束后就终止,不论子进程是否执行完;

2、守护进程内无法再开启子进程,否则抛出异常。若想子进程可继续开启子进程,则需将主进程设置为非守护进程,即deamon=False。若重写继承Process类的话则可参考 Python Multiprocessing 将进程设置为非守护进程;

如果我们有两个任务需要并发执行,那么开一个主进程和一个子进程分别去执行就ok了;
如果子进程的任务在主进程任务结束后就没有存在的必要了,那么该子进程应该在开启前就被设置成守护进程;
主进程代码运行结束,守护进程随即终止;

参考

Python多进程编程
多进程和守护进程 (推荐)

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 158,117评论 4 360
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 66,963评论 1 290
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 107,897评论 0 240
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,805评论 0 203
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,208评论 3 286
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,535评论 1 216
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,797评论 2 311
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,493评论 0 197
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,215评论 1 241
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,477评论 2 244
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 31,988评论 1 258
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,325评论 2 252
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 32,971评论 3 235
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,055评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,807评论 0 194
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,544评论 2 271
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,455评论 2 266

推荐阅读更多精彩内容

  • 1.进程和线程 1.1系统多任务机制 多任务操作机制的引入主要是在相同的硬件资源下怎么提高任务处理效率的!多任务的...
    _宁采臣阅读 934评论 0 6
  • 一、进程的概念 相信很多同学都听说过windows、linux,MacOS都是多任务,多用户的操作系统。那什么是多...
    转身后的那一回眸阅读 924评论 0 1
  • 多进程 要让python程序实现多进程,我们先了解操作系统的相关知识。 Unix、Linux操作系统提供了一个fo...
    蓓蓓的万能男友阅读 571评论 0 1
  • 进程是正在执行中的应用程序,一个进程包含了该应用程序的所有信息。计算机中多线程的操作已经可以实现多人物的处理机制了...
    阿猫阿狗py阅读 1,596评论 0 0
  • 本命年最初为五行数命回归之年,即60年一本命年,目前主要是十二年一遇的农历属相所在的年份,俗称属相年。在传统习俗中...
    简之繁阅读 300评论 0 0