How do I stack arrays in NumPy?

How do I stack arrays in NumPy?

You can use the numpy.vstack() function to stack arrays vertically or numpy.hstack() for horizontal stacking.

Can you show me an example of using vstack?

Sure! Here's a simple example:

import numpy as np

array1 = np.array([[1, 2], [3, 4]])
array2 = np.array([[5, 6]])
result = np.vstack((array1, array2))
print(result)

This will stack array1 and array2 vertically.

What if the arrays have different shapes?

If the arrays have different shapes, you need to ensure they are compatible for stacking. For example, the number of columns must match.

Is there a way to stack them along a new axis?

Yes! You can use numpy.stack() to stack arrays along a new axis. For example, np.stack((array1, array2), axis=0) will create a new dimension.