How can I reshape an array using NumPy in Python?

How can I reshape an array using NumPy in Python?

You can use the reshape() function in NumPy to change the shape of an array. For example, if you have a 1D array with 6 elements, you can reshape it to a 2D array with 2 rows and 3 columns.

Can you show me a code example?

Sure! Here's a simple example:

import numpy as np

# Create a 1D array
arr = np.array([1, 2, 3, 4, 5, 6])

# Reshape it to 2D
reshaped_arr = arr.reshape(2, 3)
print(reshaped_arr)

This will output:

[[1 2 3]
 [4 5 6]]

What if I want to reshape it to a different size?

You can reshape it to any size as long as the total number of elements remains the same. For example, you could reshape it to (3, 2) like this:

reshaped_arr = arr.reshape(3, 2)

Are there any limitations I should be aware of?

Yes, the new shape must have the same number of elements as the original array. If you try to reshape it to a size that doesn't match, NumPy will raise an error.