python多处理共享内存数组不允许二维数组

Question

我想将二维数组存储为共享内存数组，因此多处理模块创建的每个进程都可以访问二维数组。但似乎只允许使用矢量。如果我将以下示例中的形状从 (6,1) 更改为 (2,3)，则示例将不起作用，并显示“TypeError：只有长度为 1 的数组可以转换为 Python 标量”

from multiprocessing import Process, Value, Array
import numpy as np

def f(n,a):
    a[:] = np.ones(shape=(6,1), dtype=float)[:]

if __name__ == '__main__':
    tmp =  np.zeros(shape=(6,1), dtype=float)
    print tmp

    arr = Array('f', tmp)

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

    print 'done'
    print arr[:]

Answer 1

这是因为mutiprocessing.Array 是python 的array 类型的包装器，而不是numpy.ndarray 和python 的array 类型不支持多维。具体来说，请查看initializer 的文档：

一个新数组，其项目受类型代码限制，并从可选的初始化值初始化，该值必须是列表、字符串或可迭代适当类型的元素。

你有一个可迭代的（一个多维数组），但它产生视图/数组，而不是适当类型的元素。