ENH: add an axis argument to searchsorted