[python] Find indices of elements equal to zero in a NumPy array