Array indexing in numpy,"a[([i for i in range(a.shape[0]) if i != 1]), :, :]"