NumPy⑯ （行列要素同士の加算・減算）

行列要素同士の加算・減算

NumPy で、行列のすべての要素に対して演算を行う仕組みを ブロードキャスト と言います。

行列に対してスカラー演算を行うと、ブロードキャスト の仕組みによって全ての要素に同じ演算が適用されます。

この ブロードキャスト の仕組みを使って、行列の要素同士の足し算・引き算を行うことができます。

サンプルコード

まず２×２の行列を２つ定義して、足し算を行います。

[Google Colaboratory]

1
2
3
4
5
6
7
8
import numpy as np
a = np.array([[1, 2],   # 2×2の行列を作成
              [3, 4]]
             )
b = np.array([[4, 3],   # 2×2の行列を作成
              [2, 1]]
            )
a + b

[実行結果]

次に同じ行列を使って、引き算を行います。

[Google Colaboratory]

1
a - b

[実行結果]

行列の要素同士 で 足し算 と 引き算 を行うことができました。