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

投稿者: | 2014年6月23日

例として、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.]])

コメントを残す

メールアドレスが公開されることはありません。 * が付いている欄は必須項目です

*