python - 在 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？

标签： pythonnumpy

Answer 1

您应该在 numpy 包中使用 rollaxis 函数。示例代码如下：

import numpy as np
aa = np.arange(30).reshape([5, 6])
print(numpy.rollaxis(a, 0)[1])

它打印出第一个轴上的第二个元素。