Python: slicing a multi-dimensional array,"slice = [arr[i][0:2] for i in range(0, 2)]"