多重のリスト内包表記で並列計算を行いたい

多重のリスト内包表記で並列計算を行いたいと考えています。
具体的には簡単な例ではありますが、以下のような２重のリストの計算をmultiprocessingで並列化したいです。

python
1[[x + y
2  for y in range(2, 7, 2)]
3  for x in range(3, 5, 1)]

以下のプログラムのように、１次元リストとして出力することは可能ですが、できれば２次元リストとして取得したいと考えています。

python
1
2import numpy as np
3from multiprocessing    import Pool
4
5def main(x, y):
6    return x+ y
7
8def wrapper(args):
9    return main(*args)
10
11def multi_process(sampleList):
12    # Number of processes: 12 (parallel processing with 12 cpu)
13    p = Pool(12)
14    output = p.map(wrapper, sampleList)
15    p.close() # Terminate the process
16    return output
17
18sampleList = [
19    (x, y) 
20     for x      in np.arange(2, 7, 2)
21     for y      in np.arange(3, 5, 1)
22  ]
23allinfo = multi_process(sampleList)
24
25allinfo
26
27# [5, 6, 7, 8, 9, 10]
28

どのようにしたらよろしいでしょうか。

行動規範の内容に同意します

回答1件

自己解決

numpy の reshape 関数を利用することで、行いたい操作ができました。

python
1import numpy as np
2from multiprocessing    import Pool
3
4X = [ x for x in np.arange(2, 7, 2)]
5Y = [ y for y in np.arange(3, 5, 1)]
6
7def main(x, y):
8    return x+ y
9
10def wrapper(args):
11    return main(*args)
12
13def multi_process(sampleList):
14    # Number of processes: 12 (parallel processing with 12 cpu)
15    p = Pool(12)
16    output = p.map(wrapper, sampleList)
17    p.close() # Terminate the process
18    return output
19
20sampleList = [(x, y) 
21     for x  in X
22     for y  in Y]
23
24allinfo = multi_process(sampleList)
25allinfo = np.array(allinfo)
26allinfo.reshape(len(X), len(Y))
27
28
29# array([[ 5,  6],
30#        [ 7,  8],
31#        [ 9, 10]])
32
33