spekk.util.slicing.slice_array_1

Contents

spekk.util.slicing.slice_array_1#

spekk.util.slicing.slice_array_1(arr: Sliceable, axis: int, indices: int | slice | Sequence[int] | None)[source]#

Select the indices along the given axis of an array.

>>> import numpy as np
>>> arr = np.array([[1,2,3], [4,5,6]])
>>> slice_array_1(arr, 0, 0)
array([1, 2, 3])
>>> slice_array_1(arr, 1, [0, 2])
array([[1, 3],
       [4, 6]])
>>> slice_array_1(arr, None, [0, 2])
array([[1, 2, 3],
       [4, 5, 6]])
Contents