What are some examples of array broadcasting in NumPy?

What are some examples of array broadcasting in NumPy?

Array broadcasting in NumPy allows operations on arrays of different shapes. For example, if you have a 1D array and a 2D array, NumPy can automatically expand the dimensions of the 1D array to match the 2D array's shape.

Can you show me a specific example of this?

Sure! If you have a 1D array a = np.array([1, 2, 3]) and a 2D array b = np.array([[10, 20, 30], [40, 50, 60]]), adding them together will broadcast a to [[1, 2, 3], [1, 2, 3]], resulting in [[11, 22, 33], [41, 52, 63]].

What happens if the shapes are not compatible?

If the shapes are not compatible for broadcasting, NumPy will raise a ValueError. For example, trying to add a shape (3,) array to a shape (2, 3) array will fail.

Are there any other operations that support broadcasting?

Yes, broadcasting supports various operations like addition, subtraction, multiplication, and division, making it very versatile for array manipulations.