DOC: document improvements to `np.digitize`