NumPy - Intro to Python

ndarray型（行列）から任意の列のみ抜き出す

：（コロン）と，（カンマ）を組み合わせて指定をする。以下は、それぞれ1列目、2列目のみを抜き出した例。

>>> import numpy as np
>>> d = np.array([[1, 2, 3], [4, 5, 6]])
>>> d
array([[1, 2, 3],
       [4, 5, 6]])
>>> d1 = d[:, 0]
>>> d2 = d[:, 1]
>>> print(d1)
[1 4]
>>> print(d2)
[2 5]