Numpy:ブロードキャスト不可の足し算をループ文を使わずに計算する。

前提・実現したいこと

a(shape=(4, 2)) の配列の0次元の各値にb(shape=(5, 2))をブロードキャストで足し算したい。

Python
1import numpy as np
2
3a = np.arange(10).reshape(5, 2)
4b = np.arange(8).reshape(4, 2)
5
6result #求めている結果([a+b[0], a+b[1], a+b[2], a+b[3]])
7array([[[ 0,  2],
8        [ 2,  4],
9        [ 4,  6],
10        [ 6,  8],
11        [ 8, 10]],
12
13       [[ 2,  4],
14        [ 4,  6],
15        [ 6,  8],
16        [ 8, 10],
17        [10, 12]],
18
19       [[ 4,  6],
20        [ 6,  8],
21        [ 8, 10],
22        [10, 12],
23        [12, 14]],
24
25       [[ 6,  8],
26        [ 8, 10],
27        [10, 12],
28        [12, 14],
29        [14, 16]]])

発生している問題・エラーメッセージ

import numpy as np
a = np.arange(10).reshape(5, 2)
b = np.arange(8).reshape(4, 2)
a+b
#ValueError: operands could not be broadcast together with shapes (5,2) (4,2)

補足情報（FW/ツールのバージョンなど）

Python 3.9.2
Numpy 1.20.2
Windows 10

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

回答1件

ベストアンサー

shapeを変えないと無理だと思います。
shapeを変えて良いなら以下です。

python
1>>> a.reshape(5,1,2)+b
2array([[[ 0,  2],
3        [ 2,  4],
4        [ 4,  6],
5        [ 6,  8]],
6
7       [[ 2,  4],
8        [ 4,  6],
9        [ 6,  8],
10        [ 8, 10]],
11
12       [[ 4,  6],
13        [ 6,  8],
14        [ 8, 10],
15        [10, 12]],
16
17       [[ 6,  8],
18        [ 8, 10],
19        [10, 12],
20        [12, 14]],
21
22       [[ 8, 10],
23        [10, 12],
24        [12, 14],
25        [14, 16]]])