How can I use NumPy for conditional selection in Python?

How can I use NumPy for conditional selection in Python?

You can use boolean indexing in NumPy for conditional selection. For example, if you have an array, you can select elements based on a condition.

Can you show me an example of that?

Sure! If you have an array arr = np.array([1, 2, 3, 4, 5]), you can select elements greater than 3 with arr[arr > 3], which will return [4, 5].

What if I want to select elements that are even?

You can use arr[arr % 2 == 0] to select even elements. For the array arr, this will return [2, 4].

Is there a way to modify the selected elements?

Yes! You can modify the selected elements directly. For example, arr[arr > 3] = 0 will set all elements greater than 3 to 0.