Question

Let’s say I have the following Numpy array:

array([[3, 5, 0], [7, 0, 2]])

I now want to add 2 where the value is not 0. What would be the fastest way to do this? I have to manipulate quite large multidimensional arrays?

Answer 1

It seems to me that:

a[a!=0] += 2

should work.

(for the limited case of testing for non-zero-ness), you might be able to speed things up with (you’d need to timeit to see):

mask = a.astype(bool)
a[mask] += 2

---

Of course, you can save yourself the mask calculation if you can reuse the same mask at different spots (which is a pretty restrictive constraint):

mask = a != 0
a[mask] += 2
#some more code ...
a[mask] *= 3
#more code ...

---

Of course, If this is enough of a bottleneck, you can always write a little C/Fortran extension to do this for you (using Cython or f2py respectively). This would avoid the overhead of mask creation.