numpy：将（n，）数组转换为（n，1）数组的语法/习惯用法？

Question

我想把一个numpyndarray形状（n，）的对象转换成一个具有形状（n，1）的对象。我想到的最好方法是运行我自己的\u to\u col函数：

def _to_col(a):
    return a.reshape((a.size, 1))

但我很难相信，这样一个无处不在的操作还没有内置到numpy的语法中。我想我只是没能找到正确的谷歌搜索找到它。在

Answer 1

在扩展尺寸是我最喜欢的当我想添加任意轴。在

无或新轴适用于不需要灵活轴的代码。（艾克斯的回答）

>>> np.expand_dims(np.arange(5), 0).shape
(1, 5)
>>> np.expand_dims(np.arange(5), 1).shape
(5, 1)

示例用法：按任意给定轴对数组进行降级

^{pr2}$

Answer 2

我会使用以下方法：

a[:,np.newaxis]

另一种写同样东西的方法是：

^{pr2}$

以上所有操作（包括您的版本）都是恒定时间操作。在