【NumPy】行列の内積 numpy.dot

この記事では、Python言語とNumPyを用いて、行列の内積を求める方法をソースコード付きで解説します。

行列の内積

PythonとNumPyの「numpy.dot(A, B)」を利用することで、行列A, Bの内積を求めることが出来ます。

ソースコード

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

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


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

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

    C = np.dot(A,B)             # 行列AとBの内積を計算
    print(C)


if __name__ == '__main__':
    main()

実行結果

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

[[  8.  11.  12.]
 [  7.  10.  12.]
 [  6.   8.   9.]]

おすすめ記事

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

コメント