ndindex_numpy_slice_creation_canonicalization_and_intersection.py

1import numpy as np
2from ndindex import ndindex, Slice
3
4# Create an index object
5idx = ndindex(slice(0, 10, 2))
6print(f"Original index: {idx}")
7
8# Canonicalize an index against a specific array shape
9# This handles negative steps, out of bounds, and None (newaxis)
10shape = (15,)
11canonical = idx.reduce(shape)
12print(f"Canonical index for shape {shape}: {canonical}")
13
14# Use the index to slice a NumPy array
15a = np.arange(15)
16print(f"Sliced array: {a[idx.raw]}")
17
18# Perform index arithmetic: find the intersection of two slices
19s1 = Slice(0, 10)
20s2 = Slice(5, 15)
21intersection = s1.as_subindex(s2)
22print(f"Intersection of {s1} and {s2}: {intersection}")