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

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

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

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

ソースコード

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

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

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

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

関連記事