Python Numpy 3d Array Slicing In Java