`np.nonzero` surely ... :wink: 

Didn't know it existed. Live and learn. Thanks!