【NumPy】行列(配列)の対角成分を取得 (numpy.diag)

この記事では、Python言語とNumPyを用いて行列(配列)の対角成分を取得する方法をソースコード付きで解説します。

行列(配列)の対角成分を取得

Pythonの数値計算モジュールNumPyでは、numpy.diag(array)メソッドで配列arrayの対角成分を取り出すことができます。

ソースコード

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

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

def main():
    # 配列の宣言・初期化
    A = np.array([[1, 2, 3],
                  [4, 5, 6],
                  [7, 8, 9]])
    # 画面出力
    print(np.diag(A))

if __name__ == '__main__':
    main()

実行結果

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

[1 5 9]

【おすすめ記事】
Python入門 サンプル集
NumPy入門 サンプル集

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