numpyでデータを扱う その4: インデックスとスライス

# numpyのインストール
In [1]: import numpy as np

# 1次元配列を作成
In [2]: a = np.array([1,2,3])

# 要素の1番目を取得
In [3]: a[0]
Out[3]: 1

# 要素の2番目以降を取得
In [4]: a[1:]
Out[4]: array([2, 3])

# 要素の最後を取得
In [5]: a[-1]
Out[5]: 3

# 2次元配列を作成
In [7]: b = np.array([[1,2,3],[4,5,6]])

# 配列の確認
In [8]: b
Out[8]:
array([[1, 2, 3],
       [4, 5, 6]])

# 2次元配列の行方向に対して配列の取得
In [9]: b[0]
Out[9]: array([1, 2, 3])

# 2行目の1要素目を取得
In [10]: b[1,0]
Out[10]: 4

# 全行の3要素目を取得
In [11]: b[:,2]
Out[11]: array([3, 6])

# 2行目の全要素を取得
In [12]: b[1,:]
Out[12]: array([4, 5, 6])

# 1行目の2要素目以降を取得
In [13]: b[0,1:]
Out[13]: array([2, 3])

# 全行の1要素目と3要素目を取得
In [14]: b[:,[0,2]]
Out[14]:
array([[1, 3],
       [4, 6]])