从生成的多处理obj中提取矩阵

Question

以下代码生成0,1个值的矩阵：

def func(num):
    X = [random.randint(0, 2 ** 16) for _ in range(num)]
    X = list(set(X))
    X = [('{0:0' + str(16) + 'b}').format(x) for x in X]
    X = np.asarray([list(map(int, list(x))) for x in X], dtype=np.int8)
    return X

mlp = multiprocessing.Pool(multiprocessing.cpu_count()-1)
X = mlp.map(func, [num])
print X
mlp.close()
mlp.join()

矩阵X具有以下输出：

[array([[1, 1, 0, ..., 1, 1, 1],
       [0, 1, 1, ..., 1, 0, 0],
       [0, 0, 0, ..., 1, 0, 0],
       ..., 
       [1, 0, 1, ..., 1, 1, 0],
       [0, 0, 1, ..., 0, 0, 0],
       [1, 0, 1, ..., 0, 0, 0]], dtype=int8)]

但我只想：

[[1, 1, 0, ..., 1, 1, 1],
[0, 1, 1, ..., 1, 0, 0],
[0, 0, 0, ..., 1, 0, 0],
..., 
[1, 0, 1, ..., 1, 1, 0],
[0, 0, 1, ..., 0, 0, 0],
[1, 0, 1, ..., 0, 0, 0]]

如何做到这一点

Answer 1

func的定义返回一个numpy数组，而不是本机python列表。如果希望从该函数中获取本机Python列表，可以在numpy数组对象上使用tolist()方法。如果func（）需要返回numpy数组而不是本机Python列表，原因是什么

import multiprocessing
import numpy as np
import random

def func(num):
    X = [random.randint(0, 2 ** 16) for _ in range(num)]
    X = list(set(X))
    X = [('{0:0' + str(16) + 'b}').format(x) for x in X]
    X = np.asarray([list(map(int, list(x))) for x in X], dtype=np.int8)
    return X

mlp = multiprocessing.Pool(multiprocessing.cpu_count()-1)
X = mlp.map(func, [10])
mlp.close()
mlp.join()

data = X[0].tolist()

print(data)