【NumPy】行列の加算・減算

この記事では、Python言語とNumPyを用いて行列の加算・減算(和・差)を求める方法をソースコード付きで解説します。

行列の加算・減算

PythonとNumPyを利用することで、行列の足し算(加算)・引き算(減算)を求めることが出来ます。

ソースコード

サンプルプログラムのソースコードです。

# -*- coding: utf-8 -*-
import numpy as np


def main():
    A = np.array([[1,0,0]   # 行列Aの生成
                 ,[0,1,0]
                 ,[0,0,1]])

    B = np.array([[2,0,0]   # 行列Bの生成
                 ,[0,2,0]
                 ,[0,0,2]])

    C = A + B               # 行列の和
    print(C)
    
    C = A - B               # 行列の差
    print(C)

if __name__ == '__main__':
    main()

実行結果

サンプルプログラムの実行結果です。

[[3 0 0]
 [0 3 0]
 [0 0 3]]

[[-1  0  0]
 [ 0 -1  0]
 [ 0  0 -1]]

おすすめ記事

Python入門 サンプル集
NumPy入門 サンプル集

シェア&フォローお願いします!