如何初始化字符串数组以进行多处理

时间:2015-09-06 02:08:01

标签: python python-3.x multiprocessing python-multithreading

以下是流程之间共享状态的示例代码

from multiprocessing import Process, Value, Array

def f(n, a):
    n.value = 3.1415927
    for i in range(len(a)):
        a[i] = -a[i]

if __name__ == '__main__':
    num = Value('d', 0.0)
    arr = Array('i', range(10))

    p = Process(target=f, args=(num, arr))
    p.start()
    p.join()

    print(num.value)
    print(arr[:])

输出

3.1415927
[0, -1, -2, -3, -4, -5, -6, -7, -8, -9]

我想用字符串元素而不是整数元素初始化列表。然后我想分配列表特定的字符串元素。我的代码如下。

from multiprocessing import Process, Value, Array

def f(a):
    a = ["up", "down", "left"]

if __name__ == '__main__':
    arr = Array('b', [])

    p = Process(target=f, args=(arr))
    p.start()
    p.join()

    print(arr[:])

我希望输出为

["up", "down", "left"]

但我获得了输出

TypeError: f() missing 1 required positional argument: 'a'
[]

2 个答案:

答案 0 :(得分:5)

你需要传递一个args元组,你需要添加一个尾随逗号来创建一个元组:

 p = Process(target=f, args=(arr,)) # <- trailing comma

或明确使用元组:

 args=tuple([arr]))

逗号创建元组而不是parens。

不知道多处理如何帮助您获得所需的输出:

from multiprocessing import Process, Value, Array
from ctypes import c_char_p
def f(a):
    a[:] = ["up", "down", "left"]


if __name__ == '__main__':
    arr = Array(c_char_p, 3)
    p = Process(target=f, args=(arr,))
    p.start()
    p.join()
    print(arr[:])
['up', 'down', 'left']

第一个arg to Array是我们在添加字符串时需要使用ctypes.c_char_p的类型,第二个arg是数组的大小,即3个元素。

答案 1 :(得分:3)

流程希望args成为元组,而只是传递arr

https://docs.python.org/2/library/multiprocessing.html

  

args 是目标调用的参数元组。

a = (arr)
print(type(a))
# Output: <class 'multiprocessing.sharedctypes.SynchronizedArray'>

a = (arr,)
print(type(a))
# Output: <type 'tuple'>

这可以解决您的问题:

p = Process(target=f, args=(arr,))  # notice the , after arr