【NumPy】2次元配列から特定の行を取り出し

この記事では、Python言語とNumPyを用いて、2次元配列から特定の行だけを取り出す方法をソースコード付きで解説します。

特定の行を抽出

Pythonの数値計算モジュールNumPyでは、次のようにして2次元配列から特定の行にある配列(1次元配列)を取り出せます。

書式

ndarray[要素番号]

■返り値
(要素番号+1)行目のデータ(1次元配列)

例えばndarray[1]なら2行目のデータを取り出せます。

ソースコード

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

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

# 2次元配列
A = np.array([[1, 2],
              [3, 4],
              [5, 6]])

# 1行目の値を取り出し

a1 = A[0]
# 結果表示
print(a1) # [1 2]

【関連記事】
Python入門 サンプル集
NumPy入門 サンプル集

関連記事