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

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

特定の行を抽出

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

書式

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[:,1]
    # 結果表示
    print(a1)  # [2 4 6]

if __name__ == "__main__":
    main()

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

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