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

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

特定の行を抽出

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

書式

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[:,1]

# 結果表示
print(a1)  # [2 4 6]

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

コメント