python - numpy:将(an，)数组转换为(n，1)数组的语法/习惯用法？

Question

我想将一个形状为 (n,) 的 numpy ndarray 对象转换为一个形状为 (n, 1) 的对象。我想出的最好办法是滚动我自己的 _to_col 函数:

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

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

Answer 1

我会使用以下内容:

a[:,np.newaxis]

另一种(但可能不太清楚)写同样东西的方法是:

a[:,None]

以上所有(包括您的版本)都是恒定时间操作。