NumPy多次元配列のインデックス指定とスライス操作

NumPy多次元配列のインデックス指定とスライス操作がややこしいので、まとめてみた。

Python対話モードにて実験開始。
まず、実験用の配列を作る。

>>> import numpy as np
>>> x = np.arange(4*3*3*3, dtype=np.int64).reshape([4, 3, 3, 3])

xは、3×3×3の三次元配列が4個ある多次元配列である。

実験を続けよう。

>>> x[1] # インデックス1(2番目)の三次元配列
array([[[27, 28, 29],
        [30, 31, 32],
        [33, 34, 35]],

       [[36, 37, 38],
        [39, 40, 41],
        [42, 43, 44]],

       [[45, 46, 47],
        [48, 49, 50],
        [51, 52, 53]]])

参考文献

PythonとJavaScriptではじめるデータビジュアライゼーション