How do I find the dimensions of a NumPy array?

How do I find the dimensions of a NumPy array?

You can use the .shape attribute of a NumPy array to find its dimensions. For example, array.shape will return a tuple representing the size of each dimension.

Can you show me an example?

Sure! Here's a quick example:

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

This means the array has 2 rows and 3 columns.

What if I have a 3D array?

For a 3D array, you can still use the .shape attribute. For example:

array_3d = np.random.rand(2, 3, 4)
print(array_3d.shape)  # Output: (2, 3, 4)

This indicates the array has 2 matrices, each with 3 rows and 4 columns.

Is there a way to get the total number of elements?

Yes! You can use the .size attribute to get the total number of elements in the array. For example, array.size will give you the total count.