如何更改数组切片的轴？

Question

我对numpy数组还很陌生，所以任何帮助将不胜感激。

我想沿第二个轴获取n x m数组的单个切片，结果是一个n x 1数组，例如

a = np.array([[1, 2, 3],
              [4, 5, 6]])

然后我想要：

some_function(a, 0) = array([[1], [4]])  # to get slice of a, along index 0

我尝试过a[:, 0]给出array([1, 4])的情况。

并且：

np.transpose(a[:, 0])

还给出：

array([1, 4])

哪个让我感到困惑

我确定这确实很简单，但是找不到正确的some_function！

Answer 1

所以我用np.reshape解决了它：

some_function（a，0）= np.reshape（a [：，0]，（2,1））

但这似乎不太优雅。有人得到了更整洁的解决方案吗？