What are some examples of vectorization in NumPy?

What are some examples of vectorization in NumPy?

Vectorization in NumPy refers to the process of applying operations to entire arrays instead of individual elements. For example, if you want to add two arrays element-wise, you can do it directly without using loops.

Can you show me a code example?

Sure! Here's a simple example:

import numpy as np

a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
result = a + b  # This adds the two arrays element-wise
print(result)  # Output: [5 7 9]

What are the benefits of using vectorization?

The benefits of vectorization include improved performance, as operations are executed in compiled code, and cleaner, more readable code since you avoid explicit loops.

Are there any limitations to vectorization?

Yes, vectorization can be less efficient for very small arrays due to overhead, and not all operations can be vectorized, especially those that depend on previous results.