[英]Concatenating 2 dimensional numpy arrays in Python
我想连接这两个 arrays
a = np.array([[1,2,3],[3,4,5],[6,7,8]])
b = np.array([9,10,11])
这样
a = [[1,2,3,9],[3,4,5,10],[6,7,8,11]]
尝试使用连接
for i in range(len(a)):
a[i] = np.concatenate(a[i],[b[i]])
出现错误:
TypeError: 'list' object cannot be interpreted as an integer
尝试使用 append
for i in range(len(a)):
a[i] = np.append(a[i],b[i])
得到另一个错误:
ValueError: could not broadcast input array from shape (4,) into shape (3,)
(stackoverflow 的新手,抱歉,如果我没有很好地格式化这个)
您可以为此使用hstack
和矢量广播:
a = np.array([[1,2,3],[3,4,5],[6,7,8]])
b = np.array([9,10,11])
res = np.hstack((a, b[:,None]))
print(res)
Output:
[[ 1 2 3 9]
[ 3 4 5 10]
[ 6 7 8 11]]
请注意,您不能使用concatenate
,因为数组具有不同的形状。 hstack
将多维arrays 水平堆叠,所以它只是在此处的末尾添加一个新行。 需要广播操作( b[:,None]
),以便附加向量是垂直向量。
你可以这样做:
np.append(a,b.reshape(-1,1),axis=1)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.