python - 将两个一维连接到两列数组

Question

a = np.array([1, 2, 3])
aa = np.array([1], [2], [3])
b = np.array([1, 2, 3])
bb = np.array([1], [2], [3])
np.concatenate((a, b), axis = 1)
array([1, 2, 3, 1, 2, 3])     # It's ok, that's what I was expecting
np.concatenate((a, b), axis = 0)
array([1, 2, 3, 1, 2, 3])     # It's not ok, that's not what I was expecting

我期待：

array([[1, 2, 3],
       [1, 2, 3]])

即使使用 aa 和 bb，我也会得到同样的不一致。那么是否有一个简单的解决方案可以沿轴 0 连接两个一维数组？

Answer 1

注意a和b都是一维的；没有轴 1 可以连接。你想要vstack：

>>> import numpy as np
>>> a = np.array([1,2,3])
>>> b = a.copy()
>>> np.vstack([a,b])
array([[1, 2, 3],
       [1, 2, 3]])

或者，您可以重塑a并b首先：

>>> np.concatenate([a[np.newaxis,:],b[np.newaxis,:]],axis = 0)
array([[1, 2, 3],
       [1, 2, 3]])