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

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

特定の行を抽出

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

書式

ndarray[要素番号]

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

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

ソースコード

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

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

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

    # 1行目の値を取り出し
    a1 = A[0]
    # 結果表示
    print(a1) # [1 2]

if __name__ == "__main__":
    main()

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

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