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

我有 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)。但我没有。

有什么想法吗？

请您参考如下方法:

你可以使用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