numpy数组中任何轴上的样本

Question

我有以下代码：

if axis == 0:
    return m[idx, :]
elif axis == 1:
    return m[:, idx]
else:
    raise('Argument axis currently accepts only values 0 and 1.')

如何针对axis的任何形状和任何值推广此行为？

Answer 1

我找到了答案：

return np.take(m, idx, axis=axis)