Method __get_sliced_result__

detrsmpl/data/data_structures/human_data.py:1404–1430

Slice input_data along slice_dim with slice_range. Args: input_data (Union[np.ndarray, list, tuple]): Data to be sliced. slice_dim (int): Dimension to be sliced. slice_range (slice): An instance of class sli

(
            cls, input_data: Union[np.ndarray, list, tuple], slice_dim: int,
            slice_range: slice)

    @classmethod
    def __get_sliced_result__(
            cls, input_data: Union[np.ndarray, list, tuple], slice_dim: int,
            slice_range: slice) -> Union[np.ndarray, list, tuple]:
        """Slice input_data along slice_dim with slice_range.

        Args:
            input_data (Union[np.ndarray, list, tuple]):
                Data to be sliced.
            slice_dim (int):
                Dimension to be sliced.
            slice_range (slice):
                An instance of class slice.

        Returns:
            Union[np.ndarray, list, tuple]:
                A slice of input_data.
        """
        if isinstance(input_data, np.ndarray):
            slice_list = [
                slice(None),
            ] * len(input_data.shape)
            slice_list[slice_dim] = slice_range
            sliced_data = input_data[tuple(slice_list)]
        else:
            sliced_data = \
                input_data[slice_range]
        return sliced_data