w3resource

How to create a 3D array and print its strides using NumPy?


Write NumPy program to create a 3D array of shape (2, 3, 4) and print its strides.

Sample Solution:

Python Code:

import numpy as np

# Create a 3D array of shape (2, 3, 4)
x = np.array([[[1, 2, 3, 4], 
               [5, 6, 7, 8], 
               [9, 10, 11, 12]],

              [[13, 14, 15, 16], 
               [17, 18, 19, 20], 
               [21, 22, 23, 24]]])

# Print the strides of the array
print("Strides of the array:", x.strides)

Output:

Strides of the array: (48, 16, 4)

Explanation:

  • Import NumPy: Import the NumPy library to handle array operations.
  • Create 3D array x: Define a 3D array x with shape (2, 3, 4) containing integers.
  • Print Strides: Print the strides of the array using the strides attribute.

Python-Numpy Code Editor: