w3resource

Python Function: Slicing the third dimension of a multidimensional array

Python ellipsis (...) Data Type: Exercise-3 with Solution

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:

Flowchart: Python Function: Slicing the third dimension of a multidimensional array.

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

What is the difficulty level of this exercise?

Test your Programming skills with w3resource's quiz.



Become a Patron!

Follow us on Facebook and Twitter for latest update.

It will be nice if you may share this link in any developer community or anywhere else, from where other developers may find this content. Thanks.

https://198.211.115.131/python-exercises/extended-data-types/python-extended-data-types-index-ellipsis-exercise-3.php