OH that's how np.digitize works!