NumPy shape()

The shape() method returns the shape of an array i.e. the number of elements in each dimension.

Example

import numpy as np

array = np.array([[0, 1], [2, 3]])

# return shape of the array shapeOfArray = np.shape(array)
print(shapeOfArray) # Output : (2, 2)

shape() Syntax

The syntax of shape() is:

numpy.shape(array)

shape() Argument

The shape() method takes a single argument:

  • array - an array whose shape is to be determined

shape() Return Value

The shape() method returns the shape of an array as a tuple.


Example 1: Shape of Arrays

import numpy as np

# create 1D and 2D arrays
array1 = np.array([0, 1, 2, 3])
array2 = np.array([[8, 9]])

# shape of array1 shape1 = np.shape(array1)
print('Shape of the first array : ', shape1)
# shape of array2 shape2 = np.shape(array2)
print('Shape of the second array : ', shape2)

Output

Shape of the first array :  (4,)
Shape of the second array :  (1, 2)

Example 2: Shape of Array of Tuples

import numpy as np

# array of tuples
array1 = np.array([(0, 1), ( 2, 3)])

dataType = [('x', int), ('y', int)]

# dtype sets each tuple as an individual array element
array2 = np.array([(0, 1), ( 2, 3)], dtype = dataType)

# shape of array1 shape1 = np.shape(array1)
print('Shape of first array : ', shape1)
# shape of array2 shape2 = np.shape(array2)
print('Shape of second array : ', shape2)

Output

Shape of first array :  (2, 2)
Shape of second array :  (2, )

Here, array1 and array2 are 2-dimensional arrays with tuples as their elements. The shape of array1 is (2, 2).

However, the shape of array2 is (2, ), which is one dimensional.

This is because we've passed the dtype argument, which restricts the structure of array2.

array2 contains two elements, each of which is a tuple with two integer values, but each element is treated as a single entity with two fields x and y.

Therefore, array2 is one dimensional with two rows and one column.

Our premium learning platform, created with over a decade of experience and thousands of feedbacks.

Learn and improve your coding skills like never before.

Try Programiz PRO
  • Interactive Courses
  • Certificates
  • AI Help
  • 2000+ Challenges