Python:向 numpy 一维数组添加一列

Question

我有 2 个数组。

a = np.zeros(2)
b = np.ones(2)

我想将它们组合起来，使其输出为:

array([[0, 1],
       [0, 1]])

如果我有 a = np.zeros((2,1)) 和 b = np.zeros((2,1))，这可以用np.hstack((a,b)) 或 np.append(a,b,1)。但我没有。

有什么想法吗？

Answer 1

你可以使用np.c_

>>> a = np.zeros(2); b = np.ones(2)
>>> np.c_[a, b]
array([[0., 1.],
       [0., 1.]])

或者:

>>> np.array((a, b), order='F').T
array([[0., 1.],
       [0., 1.]])

这里order='F'确保最终结果是C-contiguous