When is numpy.where the right tool versus boolean assignment?
PICTURE THIS: ARRAY IN MEMORY
Index starts at 0. Scan once for max — O(n).
Simple meaning
numpy.where builds a new array from a condition and two value arrays, which is clear for vectorized if-then-else.
WHY — NumPy instead of guessing?
Why interviewers care about NumPy:
contrast on NumPy, not two memorised paragraphs.
the developer, then one case where picking wrong hurts.
Name the idea, why it exists, then one short example.
End with when you use it and one common pitfall.
STEPS — What happens step by step?
Before you speak the answer, walk the interviewer through these steps:
- 1numpy.where builds a new
array from a condition and two value arrays, which is clear for vectorized if-then-else.
- 2Boolean assignment mutates in
place and can be faster when you already own the buffer.
- 3Both still need matching
shapes via broadcasting, and neither replaces a join when the logic is keyed by IDs.
- 4Give an example
One tiny concrete case you can say aloud.
- 5Common mistake
What juniors usually get wrong.
- 6Close
When you pick this over the alternative.
EXAMPLE — See it in action
Here's a short line you can speak, broken into clear beats:
Note: Adapt this scaffold to your own project — keep it under 60–90 seconds.
Key takeaway
numpy.where builds a new array from a condition and two value arrays, which is clear for vectorized if-then-else. Boolean assignment mutates in place and can be faster when you already own the buffer.