(values, idx, axis)
| 40 | |
| 41 | |
| 42 | def _select_along_axis(values, idx, axis): |
| 43 | other_ind = np.ix_(*[np.arange(s) for s in idx.shape]) |
| 44 | sl = other_ind[:axis] + (idx,) + other_ind[axis:] |
| 45 | return values[sl] |
| 46 | |
| 47 | |
| 48 | def nanfirst(values, axis, keepdims=False): |