How to get shape of an array in python

Published on Aug. 22, 2023, 12:16 p.m.

To get the shape of an array in Python, you can use the shape attribute for numpy arrays or the shape method for regular Python lists. Here are some examples:

For a numpy array:

import numpy as np
my_array = np.array([[1, 2, 3], [4, 5, 6]])
print(my_array.shape)

This will output (2, 3), which indicates that the array has 2 rows and 3 columns.

For a regular Python list:

my_list = [[1, 2], [3, 4], [5, 6]]
n_rows = len(my_list)
n_cols = len(my_list[0])
print((n_rows, n_cols))

This will output (3, 2), which indicates that the list has 3 rows and 2 columns.

In both cases, the shape information is returned as a tuple, where the first element represents the number of rows and the second element represents the number of columns (or dimensions for numpy arrays with more than 2 dimensions).

Note that the shape attribute and method are read-only, so you cannot change the shape of an array or list using them.

Tags: