Python Numpy③ - 行列の要素へのアクセス

Numpy行列の要素へのアクセスを試してみます。

行列の要素へのアクセス

まずNumpyをインポートし、２行３列の行列を作成します。

import numpy as np

x = np.array([[1, 2, 3], [4, 5, 6]])

２行３列目の要素を取得するには次のように指定します。

print(x[1, 2])

添え字は0から始まるため２行目は 1、３列目は 2 と指定します。

[実行結果]

6

１行目の行をそのまま取得します。

print(x[0])

[実行結果]

[1 2 3]

２次元配列のまま１列目だけを取得します。

print(x[:, 0:1])

[実行結果]

[[1]
 [4]]

１次元配列で１列目を取得します。

print(x[:, 0])

[実行結果]

[1 4]