python中global變量釋疑

疑問

1. 為什么main中不能寫global x語句?
2. 在函數中如何修改global變量？
3. 在main中修改了global變量后，在子進程中為什么沒有效果？
4. 如何利用進程池的initializer參數（函數）修改子進程中的global變量？

代碼如下：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2018/9/1 8:21
# @Author  : dswang

from multiprocessing import Pool, current_process

VAR0 = 'module'


def f0(change=False):
    print 'in func f0, pid:{}'.format(current_process().pid)
    # time.sleep(2)
    if change:
        global VAR0
        VAR0 = 'f0'
    print VAR0


def ps_init():
    global_dct = globals()
    print "global_dct.keys():", global_dct.keys()
    print 'VAR0 in global:', global_dct.get('VAR0')
    global VAR0
    VAR0 = 'ps'
    print 'VAR0 in global:', global_dct.get('VAR0')


if __name__ == '__main__':
    VAR0 = 'main'
    pool = Pool(processes=1, initializer=ps_init)
    pool.apply(func=f0, args=())

    f0()
    f0(True)
    print 'in main, pid:{}'.format(current_process().pid)
    print 'VAR0 in main:'
    print VAR0

    print 'the end'

結果為：

global_dct.keys(): ['f0', 'VAR0', '__builtins__', '__file__', '__package__', 'current_process', '__name__', '__doc__', 'Pool', 'ps_init']
VAR0 in global: module
VAR0 in global: ps
in func f0, pid:6504
ps
in func f0, pid:6216
main
in func f0, pid:6216
f0
in main, pid:6216
VAR0 in main:
f0
the end

 

解疑

1、為什么main中不能寫global x語句?

因為 if __name__ == '__main__': 語句並不開辟新的作用域，所以main中的變量VAR0已經是在全局作用域，

再寫關鍵字global是多余的，會報錯：

SyntaxWarning: name 'VAR0' is assigned to before global declaration global VAR0.

 

2、在函數中如何修改global變量？

在函數中修改global變量，需要先聲明變量為global，如代碼16和24行中，在函數f0中的“global VAR0”語句。

 

3、在main中修改了global變量后，在子進程中為什么沒有效果？

注：6216是主進程，6504是子進程。

從運行結果中看到（結果第2行）：子進程中的VAR0並沒有因為main中的修改而改變，

依然保持module中的值“module”。當子進程中修改后，VAR0變成了“ps”，在隨后f0函數中依然是"ps"。

然后，回到main中(代碼34行)，VAR0為main中修改后的“main”。最后，main中函數f0對VAR0再次修改為“f0”，

因此，VAR0在f0和main中都變成了“f0”，見結果第9和第12行。

 

4、如何利用進程池的initializer參數（函數）修改子進程中的global變量？

參考代碼和問題3的回答。

 

總結

1. if語句不開辟新的作用域，因此不能在main中寫global關鍵字，直接修改即可；
2. 函數中要修改全局變量，需要先聲明為全局變量：global x;
3. 子進程中的全局變量取module中的值，main中修改module的全局變量不影響子進程。