NumPyで行列の要素（i行j列の数）を取得する

最終更新日

2023.02.18

NumPy で行列の要素は A[1, 2] のように取得します。

import numpy as np

A = np.array([[1, 2, 3],
              [4, 5, 6],
              [7, 8, 9]])

print(A[0, 0])  # 1
print(A[0, 1])  # 2
print(A[0, 2])  # 3
print(A[1, 0])  # 4
print(A[1, 1])  # 5
print(A[1, 2])  # 6
print(A[2, 0])  # 7
print(A[2, 1])  # 8
print(A[2, 2])  # 9

形式は数学の行列とほぼ同じですが、最初の列・行のインデックスは 0 です。

NumPy

配列