Python Function: Slicing the third dimension of a multidimensional array

Last update on December 21 2024 07:29:21 (UTC/GMT +8 hours)

Write a Python function that takes a multidimensional array and slices the first two elements from the third dimension.

Sample Solution:

Code:

import numpy as np 

def slice_third_dimension(arr):
    """
      Args:
        arr (numpy.ndarray): The input multidimensional array.

    Returns:
        numpy.ndarray: The sliced array.
    """
    return arr[..., :2]

# Example usage:
nums = np.array([[[1, 2, 3], [4, 5, 6]],
                           [[7, 8, 9], [10, 11, 12]],
                           [[13, 14, 15], [16, 17, 18]]])
print("Original multidimensional array:")
print(nums)

sliced_array = slice_third_dimension(nums)
print("\nSlices the first two elements from the said array:")
print(sliced_array)

Output:

Original multidimensional array:
[[[ 1  2  3]
  [ 4  5  6]]

 [[ 7  8  9]
  [10 11 12]]

 [[13 14 15]
  [16 17 18]]]

Slices the first two elements from the said array:
[[[ 1  2]
  [ 4  5]]

 [[ 7  8]
  [10 11]]

 [[13 14]
  [16 17]]]

In the exercise above, the "slice_third_dimension()" function takes a NumPy multidimensional array as input and uses the ellipsis (...) to slice the first two elements from the third dimension. In the resulting sliced array, only the first two elements of the third dimension are preserved.

Flowchart:

Previous: Python Function: Calling functions with flexible arguments.
Next: Python Program: Creating lists with gaps using ellipsis.