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]])