(x, axis)
| 66 | return x[idx] |
| 67 | |
| 68 | def create_slices(x, axis): |
| 69 | def flatten_extend(matrix): |
| 70 | flat_list = [] |
| 71 | for row in matrix: |
| 72 | flat_list.extend(row) |
| 73 | return flat_list |
| 74 | |
| 75 | if isinstance(x[0], list): |
| 76 | return [create_slices([x[i][j] for i in range(len(x))], axis) for j in range(len(x[0]))] |
| 77 | if ants.is_image(x[0]): |
| 78 | return flatten_extend([[xx.slice_image(axis, i) for i in range(xx.shape[axis])] for xx in x]) |
| 79 | else: |
| 80 | return x |