(23) numpyで要素数の異なる配列の計算

8,389 views

この記事は最終更新から 2426日 が経過しています。

例として、5行5列の配列 aと、5行1列の配列 bを用意する。

>>> a=np.eye(5)
>>> a
array([[ 1.,  0.,  0.,  0.,  0.],
       [ 0.,  1.,  0.,  0.,  0.],
       [ 0.,  0.,  1.,  0.,  0.],
       [ 0.,  0.,  0.,  1.,  0.],
       [ 0.,  0.,  0.,  0.,  1.]])
>>>
>>> b=np.ones([5,1])
>>> b
array([[ 1.],
       [ 1.],
       [ 1.],
       [ 1.],
       [ 1.]])

この場合、a+bは bを 5行5列に拡張した後に積算される。

>>> a+b
array([[ 2.,  1.,  1.,  1.,  1.],
       [ 1.,  2.,  1.,  1.,  1.],
       [ 1.,  1.,  2.,  1.,  1.],
       [ 1.,  1.,  1.,  2.,  1.],
       [ 1.,  1.,  1.,  1.,  2.]])